[Model] Add LlamaEmbeddingModel as an embedding Implementation of Lla…

…maModel (vllm-project#9806) Signed-off-by: Sumit Dubey <[email protected]>
sumitd2 · Nov 14, 2024 · a1d3604 · a1d3604
1 parent ae6019f
commit a1d3604
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/vllm/model_executor/models/registry.py b/vllm/model_executor/models/registry.py
@@ -95,6 +95,7 @@
     # [Text-only]
     "BertModel": ("bert", "BertEmbeddingModel"),
     "Gemma2Model": ("gemma2", "Gemma2EmbeddingModel"),
+    "LlamaModel": ("llama", "LlamaEmbeddingModel"),
     "MistralModel": ("llama", "LlamaEmbeddingModel"),
     "Qwen2ForRewardModel": ("qwen2_rm", "Qwen2ForRewardModel"),
     "Qwen2ForSequenceClassification": (