add jamba-classification model to registry

Signed-off-by: Yehoshua Cohen <[email protected]>
vllm-project · Dec 3, 2024 · 8e3a7d2 · 8e3a7d2
1 parent d4de8fd
commit 8e3a7d2
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
@@ -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"),