Skip to content

Commit d80d0c0

Browse files
mergennachinfacebook-github-bot
authored andcommitted
Fix issue caused by llama2->llama renaming
Differential Revision: D64499907
1 parent e86cb3a commit d80d0c0

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)