From 8e3a7d271e88b59c3fd7c10ee03a298631e69220 Mon Sep 17 00:00:00 2001 From: Yehoshua Cohen Date: Tue, 3 Dec 2024 11:54:07 +0200 Subject: [PATCH] add jamba-classification model to registry Signed-off-by: Yehoshua Cohen --- vllm/model_executor/models/registry.py | 1 + 1 file changed, 1 insertion(+) diff --git a/vllm/model_executor/models/registry.py b/vllm/model_executor/models/registry.py index c66fbce018a62..be5211e6ae5f6 100644 --- a/vllm/model_executor/models/registry.py +++ b/vllm/model_executor/models/registry.py @@ -108,6 +108,7 @@ "RobertaForMaskedLM": ("roberta", "RobertaEmbeddingModel"), "XLMRobertaModel": ("roberta", "RobertaEmbeddingModel"), "DeciLMForCausalLM": ("decilm", "DeciLMForCausalLM"), + "JambaForSequenceClassification": ("jamba", "JambaForSequenceClassification"), # noqa: E501 "Gemma2Model": ("gemma2", "Gemma2ForCausalLM"), "GlmForCausalLM": ("glm", "GlmForCausalLM"), "LlamaModel": ("llama", "LlamaForCausalLM"),