From d80d0c05273d486013abcd47cfe317b3840de565 Mon Sep 17 00:00:00 2001 From: Mergen Nachin Date: Wed, 16 Oct 2024 14:51:55 -0700 Subject: [PATCH] Fix issue caused by llama2->llama renaming Differential Revision: D64499907 --- examples/models/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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"),