Skip to content

Commit 0350809

Browse files
authored
Remove Falcon3 2x7B from CI (#17404)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
1 parent a6977db commit 0350809

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

tests/models/decoder_only/language/test_models.py

+3-4
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
"openbmb/MiniCPM3-4B",
3030
"Qwen/Qwen-7B-Chat",
3131
"Qwen/Qwen2.5-0.5B-Instruct",
32-
"ehristoforu/Falcon3-MoE-2x7B-Insruct",
32+
"TitanML/tiny-mixtral",
3333
]
3434

3535

@@ -83,9 +83,8 @@
8383
pytest.param("stabilityai/stablelm-3b-4e1t"), # stablelm
8484
pytest.param("bigcode/starcoder2-3b"), # starcoder2
8585
pytest.param(
86-
"ehristoforu/Falcon3-MoE-2x7B-Insruct", # mixtral
87-
marks=[pytest.mark.cpu_model,
88-
large_gpu_mark(min_gb=48)],
86+
"TitanML/tiny-mixtral", # mixtral
87+
marks=[pytest.mark.cpu_model],
8988
)
9089
])
9190
@pytest.mark.parametrize("max_tokens", [32])

tests/models/registry.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,7 @@ def check_available_online(
191191
trust_remote_code=True),
192192
"MistralForCausalLM": _HfExamplesInfo("mistralai/Mistral-7B-Instruct-v0.1"),
193193
"MixtralForCausalLM": _HfExamplesInfo("mistralai/Mixtral-8x7B-Instruct-v0.1", # noqa: E501
194-
{"falcon3": "ehristoforu/Falcon3-MoE-2x7B-Insruct"}), # noqa: E501
194+
{"tiny": "TitanML/tiny-mixtral"}), # noqa: E501
195195
"QuantMixtralForCausalLM": _HfExamplesInfo("mistral-community/Mixtral-8x22B-v0.1-AWQ"), # noqa: E501
196196
"MptForCausalLM": _HfExamplesInfo("mpt", is_available_online=False),
197197
"MPTForCausalLM": _HfExamplesInfo("mosaicml/mpt-7b"),

0 commit comments

Comments
 (0)