Skip to content

Commit 06e4907

Browse files
committed
update
1 parent 06f72c1 commit 06e4907

1 file changed

Lines changed: 8 additions & 3 deletions

File tree

tests/test_static_config.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,16 @@ def test_glm_5_1_uses_mlx_community_model():
88

99

1010
def test_qwen3_6_mxfp4_is_scheduler_supported():
11-
assert MODELS["mlx-community/Qwen3.6-27B-mxfp4"] == "mlx-community/Qwen3.6-27B-mxfp4"
11+
assert MODELS["Qwen/Qwen3.6-27B"] == "mlx-community/Qwen3.6-27B-mxfp4"
12+
assert "mlx-community/Qwen3.6-27B-mxfp4" not in MODELS
1213

1314

1415
def test_qwen3_6_mxfp4_model_info_uses_text_config(monkeypatch):
1516
def fake_load_config_only(model_name, local_files_only=False):
16-
assert model_name == "mlx-community/Qwen3.6-27B-mxfp4"
17+
assert model_name in {
18+
"Qwen/Qwen3.6-27B",
19+
"mlx-community/Qwen3.6-27B-mxfp4",
20+
}
1721
return normalize_model_config(
1822
{
1923
"model_type": "qwen3_5",
@@ -33,9 +37,10 @@ def fake_load_config_only(model_name, local_files_only=False):
3337

3438
monkeypatch.setattr(static_config, "load_config_only", fake_load_config_only)
3539

36-
model_info = get_model_info("mlx-community/Qwen3.6-27B-mxfp4")
40+
model_info = get_model_info("Qwen/Qwen3.6-27B")
3741

3842
assert model_info.num_layers == 64
43+
assert model_info.mlx_model_name == "mlx-community/Qwen3.6-27B-mxfp4"
3944
assert model_info.head_size == 256
4045
assert model_info.hidden_dim == 5120
4146
assert model_info.num_attention_heads == 24

0 commit comments

Comments
 (0)