Skip to content

Commit 3b8b28b

Browse files
mergennachinfacebook-github-bot
authored andcommitted
Fix issue caused by llama2->llama renaming (#6310)
Summary: Pull Request resolved: #6310 Reviewed By: kirklandsign Differential Revision: D64499907 fbshipit-source-id: b5ea566f65678bdba0f9a6032a1c049c5597acdc
1 parent e86cb3a commit 3b8b28b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

examples/models/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,8 @@
1616
"emformer_transcribe": ("emformer_rnnt", "EmformerRnntTranscriberModel"),
1717
"emformer_predict": ("emformer_rnnt", "EmformerRnntPredictorModel"),
1818
"emformer_join": ("emformer_rnnt", "EmformerRnntJoinerModel"),
19-
"llama2": ("llama2", "Llama2Model"),
19+
"llama2": ("llama", "Llama2Model"),
20+
"llama": ("llama", "Llama2Model"),
2021
"lstm": ("lstm", "LSTMModel"),
2122
"mobilebert": ("mobilebert", "MobileBertModelExample"),
2223
"mv2": ("mobilenet_v2", "MV2Model"),

0 commit comments

Comments
 (0)