diff --git a/examples/models/__init__.py b/examples/models/__init__.py index 81c9fc2b6dc..80f95af89e0 100644 --- a/examples/models/__init__.py +++ b/examples/models/__init__.py @@ -16,7 +16,8 @@ "emformer_transcribe": ("emformer_rnnt", "EmformerRnntTranscriberModel"), "emformer_predict": ("emformer_rnnt", "EmformerRnntPredictorModel"), "emformer_join": ("emformer_rnnt", "EmformerRnntJoinerModel"), - "llama2": ("llama2", "Llama2Model"), + "llama2": ("llama", "Llama2Model"), + "llama": ("llama", "Llama2Model"), "lstm": ("lstm", "LSTMModel"), "mobilebert": ("mobilebert", "MobileBertModelExample"), "mv2": ("mobilenet_v2", "MV2Model"),