[CI/Build] Update bamba test model location (vllm-project#18544)

hmellor · Yuqi Zhang · commit d80d1cfce6c7 · 2025-05-24T08:03:12.000Z
Signed-off-by: Harry Mellor &lt;19981378+hmellor@users.noreply.github.com&gt;
Signed-off-by: Yuqi Zhang &lt;yuqizhang@google.com&gt;
diff --git a/tests/models/language/generation/test_hybrid.py b/tests/models/language/generation/test_hybrid.py
@@ -31,7 +31,7 @@
     # not compatible with pip-compile.
     "pfnet/plamo-2-1b",
     "Zyphra/Zamba2-1.2B-instruct",
-    "hmellor/bamba-tiny-random",
+    "hmellor/tiny-random-BambaForCausalLM",
 ]
 
 # Avoid OOM
diff --git a/tests/models/registry.py b/tests/models/registry.py
@@ -124,7 +124,7 @@ def check_available_online(
     "BaichuanForCausalLM": _HfExamplesInfo("baichuan-inc/Baichuan2-7B-chat",
                                          trust_remote_code=True),
     "BambaForCausalLM": _HfExamplesInfo("ibm-ai-platform/Bamba-9B",
-                                        extras={"tiny": "hmellor/bamba-tiny-random"}),  # noqa: E501
+                                        extras={"tiny": "hmellor/tiny-random-BambaForCausalLM"}),  # noqa: E501
     "BloomForCausalLM": _HfExamplesInfo("bigscience/bloom-560m",
                                         {"1b": "bigscience/bloomz-1b1"}),
     "ChatGLMModel": _HfExamplesInfo("THUDM/chatglm3-6b",
diff --git a/tests/v1/test_oracle.py b/tests/v1/test_oracle.py
@@ -12,7 +12,7 @@
     "openai/whisper-large-v3",  # transcription
     "facebook/bart-large-cnn",  # encoder decoder
     "mistralai/Mamba-Codestral-7B-v0.1",  # mamba
-    "hmellor/bamba-tiny-random",  # hybrid
+    "hmellor/tiny-random-BambaForCausalLM",  # hybrid
     "BAAI/bge-m3",  # embedding
 ]
 

Original file line number	Diff line number	Diff line change
`@@ -31,7 +31,7 @@`
`31`	`31`	`# not compatible with pip-compile.`
`32`	`32`	`"pfnet/plamo-2-1b",`
`33`	`33`	`"Zyphra/Zamba2-1.2B-instruct",`
`34`		`- "hmellor/bamba-tiny-random",`
	`34`	`+ "hmellor/tiny-random-BambaForCausalLM",`
`35`	`35`	`]`
`36`	`36`
`37`	`37`	`# Avoid OOM`
Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@`
`12`	`12`	`"openai/whisper-large-v3", # transcription`
`13`	`13`	`"facebook/bart-large-cnn", # encoder decoder`
`14`	`14`	`"mistralai/Mamba-Codestral-7B-v0.1", # mamba`
`15`		`- "hmellor/bamba-tiny-random", # hybrid`
	`15`	`+ "hmellor/tiny-random-BambaForCausalLM", # hybrid`
`16`	`16`	`"BAAI/bge-m3", # embedding`
`17`	`17`	`]`
`18`	`18`