feat: update is_fa3_default_architecture (#5854)

This commit is contained in:
Yineng Zhang
2025-04-28 11:53:22 -07:00
committed by GitHub
parent f4a9f60cbd
commit 663037a7a0

View File

@@ -1973,6 +1973,8 @@ def is_fa3_default_architecture(hf_config):
"MixtralForCausalLM",
"Gemma2ForCausalLM",
"Gemma3ForConditionalGeneration",
"Qwen3ForCausalLM",
"Qwen3MoeForCausalLM",
}
return architectures[0] in default_archs