Skip to content

Commit e6deadd

Browse files
authored
support gpt-oss (#199)
1 parent 0972554 commit e6deadd

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

onnx_diagnostic/torch_models/hghub/hub_data.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@
5252
GPTJModel,feature-extraction
5353
GPTNeoModel,feature-extraction
5454
GPTNeoXForCausalLM,text-generation
55+
GptOssForCausalLM,text-generation
5556
GemmaForCausalLM,text-generation
5657
Gemma2ForCausalLM,text-generation
5758
Gemma3ForConditionalGeneration,image-text-to-text

0 commit comments

Comments
 (0)