Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "e7296b08da66b4c79eb43d0932a3d8628178d036"
Unverified Commit ac3d748d authored by Junichi Sato's avatar Junichi Sato Committed by GitHub
Browse files

[Model] Add LlamaEmbeddingModel as an embedding Implementation of LlamaModel (#9806)

parent 0ce7798f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
vllm/model_executor/models/registry.py
View file @ ac3d748d
...@@ -95,6 +95,7 @@ _EMBEDDING_MODELS = { ...@@ -95,6 +95,7 @@ _EMBEDDING_MODELS = {
# [Text-only] # [Text-only]
"BertModel": ("bert", "BertEmbeddingModel"), "BertModel": ("bert", "BertEmbeddingModel"),
"Gemma2Model": ("gemma2", "Gemma2EmbeddingModel"), "Gemma2Model": ("gemma2", "Gemma2EmbeddingModel"),
"LlamaModel": ("llama", "LlamaEmbeddingModel"),
"MistralModel": ("llama", "LlamaEmbeddingModel"), "MistralModel": ("llama", "LlamaEmbeddingModel"),
"Qwen2ForRewardModel": ("qwen2_rm", "Qwen2ForRewardModel"), "Qwen2ForRewardModel": ("qwen2_rm", "Qwen2ForRewardModel"),
"Qwen2ForSequenceClassification": ( "Qwen2ForSequenceClassification": (
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment