Skip to content

Commit 55b5280

Browse files
Add gemma3 onnxruntime tests
1 parent 8345410 commit 55b5280

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/onnxruntime/testing_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,7 @@
6868
"flaubert": "hf-internal-testing/tiny-random-flaubert",
6969
"flux": "optimum-internal-testing/tiny-random-flux",
7070
"gemma": "fxmarty/tiny-random-GemmaForCausalLM",
71+
"gemma3": "hf-internal-testing/tiny-random-Gemma3ForConditionalGeneration",
7172
"gpt2": "hf-internal-testing/tiny-random-GPT2LMHeadModel",
7273
"gpt_bigcode": "hf-internal-testing/tiny-random-GPTBigCodeModel",
7374
"gpt_bigcode-multi_query-False": "optimum-internal-testing/tiny-random-gpt_bigcode-multi_query-False",

0 commit comments

Comments
 (0)