Skip to content

Commit 0962594

Browse files
Add Gemma3 to test_decoder.py
1 parent 798d3ce commit 0962594

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
@@ -31,6 +31,7 @@
3131
ArceeOnnxConfig,
3232
BloomOnnxConfig,
3333
CohereOnnxConfig,
34+
Gemma3OnnxConfig,
3435
GemmaOnnxConfig,
3536
GraniteOnnxConfig,
3637
HeliumOnnxConfig,
@@ -106,6 +107,8 @@ class ORTModelForCausalLMIntegrationTest(ORTModelTestMixin):
106107
SUPPORTED_ARCHITECTURES.append("qwen2")
107108
if is_transformers_version(">=", str(GemmaOnnxConfig.MIN_TRANSFORMERS_VERSION)):
108109
SUPPORTED_ARCHITECTURES.append("gemma")
110+
if is_transformers_version(">=", str(Gemma3OnnxConfig.MIN_TRANSFORMERS_VERSION)):
111+
SUPPORTED_ARCHITECTURES.append("gemma3")
109112
if is_transformers_version(">=", str(MPTOnnxConfig.MIN_TRANSFORMERS_VERSION)):
110113
SUPPORTED_ARCHITECTURES.append("mpt")
111114
if is_transformers_version(">=", str(GraniteOnnxConfig.MIN_TRANSFORMERS_VERSION)):

0 commit comments

Comments
 (0)