update

Author: gufengc

src/parallax/utils/utils.py

@@ -303,9 +303,7 @@ def load_config_only(name: str, local_files_only: bool = False):
 def normalize_model_config(config: dict) -> dict:
     """Expose nested text model fields at the top level for VLM-style configs."""
     text_config = config.get("text_config")
-    if config.get("model_type") in {"qwen3_5", "qwen3_5_moe"} and isinstance(
-        text_config, dict
-    ):
+    if config.get("model_type") in {"qwen3_5", "qwen3_5_moe"} and isinstance(text_config, dict):
         normalized = {**config, **text_config}
         normalized["model_type"] = config["model_type"]
         normalized["architectures"] = config.get("architectures", normalized.get("architectures"))

tests/test_shard_loader.py

@@ -144,9 +144,7 @@ def test_qwen35_moe_uses_qwen35_text_args_and_sanitizer_module():
         }
     )
 
-    sanitizer_module, model_args = loader._load_mlx_lm_module_and_args(
-        "qwen3_5_moe", config
-    )
+    sanitizer_module, model_args = loader._load_mlx_lm_module_and_args("qwen3_5_moe", config)
 
     assert MODEL_CLASS_MAP["qwen3_5_moe"] == "mlx_lm.models.qwen3_5"
     assert sanitizer_module.__name__ == "mlx_lm.models.qwen3_5"