EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

feat: update is_fa3_default_architecture (#5854)

Browse Source
This commit is contained in:
Yineng Zhang
2025-04-28 11:53:22 -07:00
committed by GitHub
parent f4a9f60cbd
commit 663037a7a0
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/sglang/srt/utils.py
View File

@@ -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