Update default attn backend to fa3 for mimo model (#17419)

2026-01-20 21:22:27 +08:00
parent 23d765d1f2
commit 91a4cd8648
1 changed files with 1 additions and 0 deletions
@@ -2880,6 +2880,7 @@ def is_fa3_default_architecture(hf_config):
        "Glm4vForConditionalGeneration",
        "Glm4vMoeForConditionalGeneration",
        "Step3VLForConditionalGeneration",
        "MiMoV2FlashForCausalLM",
    }
    return architectures[0] in default_archs