[Model] GLM adaptation (#34124)

2026-02-09 17:32:52 +08:00
parent 5a5c43511a
commit 978a37c823
7 changed files with 13 additions and 3 deletions
--- a/tests/models/test_initialization.py
+++ b/tests/models/test_initialization.py
@@ -97,7 +97,7 @@ def can_initialize(
            "pickle error when loading `transformers.models.auto.CONFIG_MAPPING`"
        )

-    if model_arch == "DeepseekV32ForCausalLM":
+    if model_arch in ["DeepseekV32ForCausalLM", "GlmMoeDsaForCausalLM"]:
        from vllm.platforms import current_platform

        capability = current_platform.get_device_capability()