Disable gemma3 for SWA due to CUDA illegal memory access error (#8895)
Signed-off-by: Xinyuan Tong <xinyuantong.cs@gmail.com>
This commit is contained in:
@@ -2037,6 +2037,8 @@ class ServerArgs:
|
|||||||
|
|
||||||
if model_arch in [
|
if model_arch in [
|
||||||
"Gemma2ForCausalLM",
|
"Gemma2ForCausalLM",
|
||||||
|
"Gemma3ForCausalLM",
|
||||||
|
"Gemma3ForConditionalGeneration",
|
||||||
"Gemma3nForCausalLM",
|
"Gemma3nForCausalLM",
|
||||||
"Gemma3nForConditionalGeneration",
|
"Gemma3nForConditionalGeneration",
|
||||||
]:
|
]:
|
||||||
|
|||||||
Reference in New Issue
Block a user