Fix failing tests

Signed-off-by: DarkLight1337 <[email protected]>
vllm-project · Dec 19, 2024 · eb111b4 · eb111b4
1 parent cc03793
commit eb111b4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/models/registry.py b/tests/models/registry.py
@@ -138,6 +138,7 @@ class _HfExamplesInfo:
     "BertModel": _HfExamplesInfo("BAAI/bge-base-en-v1.5"),
     "Gemma2Model": _HfExamplesInfo("BAAI/bge-multilingual-gemma2"),
     "GritLM": _HfExamplesInfo("parasail-ai/GritLM-7B-vllm"),
+    "JambaForSequenceClassification": _HfExamplesInfo("ai21labs/Jamba-tiny-reward-dev"),  # noqa: E501
     "LlamaModel": _HfExamplesInfo("llama", is_available_online=False),
     "MistralModel": _HfExamplesInfo("intfloat/e5-mistral-7b-instruct"),
     "Qwen2Model": _HfExamplesInfo("ssmits/Qwen2-7B-Instruct-embed-base"),