Unverified Commit 663037a7 authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

feat: update is_fa3_default_architecture (#5854)

parent f4a9f60c
...@@ -1973,6 +1973,8 @@ def is_fa3_default_architecture(hf_config): ...@@ -1973,6 +1973,8 @@ def is_fa3_default_architecture(hf_config):
"MixtralForCausalLM", "MixtralForCausalLM",
"Gemma2ForCausalLM", "Gemma2ForCausalLM",
"Gemma3ForConditionalGeneration", "Gemma3ForConditionalGeneration",
"Qwen3ForCausalLM",
"Qwen3MoeForCausalLM",
} }
return architectures[0] in default_archs return architectures[0] in default_archs
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment