Skip to content

Commit 41f0c75

Browse files
Add Gemma3 to test_decoder.py
1 parent 510c2b7 commit 41f0c75

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/onnxruntime/test_decoder.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
BloomOnnxConfig,
3333
CohereOnnxConfig,
3434
DeepSeekV3OnnxConfig,
35+
Gemma3OnnxConfig,
3536
GemmaOnnxConfig,
3637
GraniteOnnxConfig,
3738
HeliumOnnxConfig,
@@ -108,6 +109,8 @@ class ORTModelForCausalLMIntegrationTest(ORTModelTestMixin):
108109
SUPPORTED_ARCHITECTURES.append("qwen2")
109110
if is_transformers_version(">=", str(GemmaOnnxConfig.MIN_TRANSFORMERS_VERSION)):
110111
SUPPORTED_ARCHITECTURES.append("gemma")
112+
if is_transformers_version(">=", str(Gemma3OnnxConfig.MIN_TRANSFORMERS_VERSION)):
113+
SUPPORTED_ARCHITECTURES.append("gemma3")
111114
if is_transformers_version(">=", str(MPTOnnxConfig.MIN_TRANSFORMERS_VERSION)):
112115
SUPPORTED_ARCHITECTURES.append("mpt")
113116
if is_transformers_version(">=", str(NemotronOnnxConfig.MIN_TRANSFORMERS_VERSION)):

0 commit comments

Comments
 (0)