Skip to content

GitLab

Unverified Commit 7489ec0b authored by Harry Mellor's avatar Harry Mellor Committed by GitHub
Browse files

Remove Bamba 9B from CI (#17407) 


Signed-off-by: default avatarHarry Mellor <19981378+hmellor@users.noreply.github.com>
parent 70788bdb
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
tests/models/decoder_only/language/test_hybrid.py
View file @ 7489ec0b
...@@ -28,7 +28,7 @@ HYBRID_MODELS = [ ...@@ -28,7 +28,7 @@ HYBRID_MODELS = [
# not compatible with pip-compile. # not compatible with pip-compile.
"pfnet/plamo-2-1b", "pfnet/plamo-2-1b",
"Zyphra/Zamba2-1.2B-instruct", "Zyphra/Zamba2-1.2B-instruct",
"ibm-ai-platform/Bamba-9B", "hmellor/bamba-tiny-random",
] ]
# Avoid OOM # Avoid OOM
......
tests/models/registry.py
View file @ 7489ec0b
...@@ -120,7 +120,8 @@ _TEXT_GENERATION_EXAMPLE_MODELS = { ...@@ -120,7 +120,8 @@ _TEXT_GENERATION_EXAMPLE_MODELS = {
trust_remote_code=True), trust_remote_code=True),
"BaichuanForCausalLM": _HfExamplesInfo("baichuan-inc/Baichuan2-7B-chat", "BaichuanForCausalLM": _HfExamplesInfo("baichuan-inc/Baichuan2-7B-chat",
trust_remote_code=True), trust_remote_code=True),
"BambaForCausalLM": _HfExamplesInfo("ibm-ai-platform/Bamba-9B"), "BambaForCausalLM": _HfExamplesInfo("ibm-ai-platform/Bamba-9B",
extras={"tiny": "hmellor/bamba-tiny-random"}), # noqa: E501
"BloomForCausalLM": _HfExamplesInfo("bigscience/bloom-560m", "BloomForCausalLM": _HfExamplesInfo("bigscience/bloom-560m",
{"1b": "bigscience/bloomz-1b1"}), {"1b": "bigscience/bloomz-1b1"}),
"ChatGLMModel": _HfExamplesInfo("THUDM/chatglm3-6b", "ChatGLMModel": _HfExamplesInfo("THUDM/chatglm3-6b",
......
tests/v1/test_oracle.py
View file @ 7489ec0b
...@@ -12,7 +12,7 @@ UNSUPPORTED_MODELS_V1 = [ ...@@ -12,7 +12,7 @@ UNSUPPORTED_MODELS_V1 = [
"openai/whisper-large-v3", # transcription "openai/whisper-large-v3", # transcription
"facebook/bart-large-cnn", # encoder decoder "facebook/bart-large-cnn", # encoder decoder
"mistralai/Mamba-Codestral-7B-v0.1", # mamba "mistralai/Mamba-Codestral-7B-v0.1", # mamba
"ibm-ai-platform/Bamba-9B", # hybrid "hmellor/bamba-tiny-random", # hybrid
"BAAI/bge-m3", # embedding "BAAI/bge-m3", # embedding
] ]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment