Skip to content

Commit ca86a7c

Browse files
authored
[CI/Build] Update bamba test model location (#18544)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
1 parent a35a494 commit ca86a7c

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

tests/models/language/generation/test_hybrid.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@
3131
# not compatible with pip-compile.
3232
"pfnet/plamo-2-1b",
3333
"Zyphra/Zamba2-1.2B-instruct",
34-
"hmellor/bamba-tiny-random",
34+
"hmellor/tiny-random-BambaForCausalLM",
3535
]
3636

3737
# Avoid OOM

tests/models/registry.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ def check_available_online(
124124
"BaichuanForCausalLM": _HfExamplesInfo("baichuan-inc/Baichuan2-7B-chat",
125125
trust_remote_code=True),
126126
"BambaForCausalLM": _HfExamplesInfo("ibm-ai-platform/Bamba-9B",
127-
extras={"tiny": "hmellor/bamba-tiny-random"}), # noqa: E501
127+
extras={"tiny": "hmellor/tiny-random-BambaForCausalLM"}), # noqa: E501
128128
"BloomForCausalLM": _HfExamplesInfo("bigscience/bloom-560m",
129129
{"1b": "bigscience/bloomz-1b1"}),
130130
"ChatGLMModel": _HfExamplesInfo("THUDM/chatglm3-6b",

tests/v1/test_oracle.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
"openai/whisper-large-v3", # transcription
1313
"facebook/bart-large-cnn", # encoder decoder
1414
"mistralai/Mamba-Codestral-7B-v0.1", # mamba
15-
"hmellor/bamba-tiny-random", # hybrid
15+
"hmellor/tiny-random-BambaForCausalLM", # hybrid
1616
"BAAI/bge-m3", # embedding
1717
]
1818

0 commit comments

Comments
 (0)