Skip to content

Commit cde11ed

Browse files
Add gemma3 onnxruntime tests
1 parent 82bb6cc commit cde11ed

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
@@ -67,6 +67,7 @@
6767
"flaubert": "hf-internal-testing/tiny-random-flaubert",
6868
"flux": "optimum-internal-testing/tiny-random-flux",
6969
"gemma": "fxmarty/tiny-random-GemmaForCausalLM",
70+
"gemma3": "hf-internal-testing/tiny-random-Gemma3ForConditionalGeneration",
7071
"gpt2": "hf-internal-testing/tiny-random-GPT2LMHeadModel",
7172
"gpt_bigcode": "hf-internal-testing/tiny-random-GPTBigCodeModel",
7273
"gpt_bigcode-multi_query-False": "optimum-internal-testing/tiny-random-gpt_bigcode-multi_query-False",

0 commit comments

Comments
 (0)