diff --git a/onnx_diagnostic/torch_models/hghub/hub_data.py b/onnx_diagnostic/torch_models/hghub/hub_data.py index 908cb22..517f434 100644 --- a/onnx_diagnostic/torch_models/hghub/hub_data.py +++ b/onnx_diagnostic/torch_models/hghub/hub_data.py @@ -52,6 +52,7 @@ GPTJModel,feature-extraction GPTNeoModel,feature-extraction GPTNeoXForCausalLM,text-generation + GptOssForCausalLM,text-generation GemmaForCausalLM,text-generation Gemma2ForCausalLM,text-generation Gemma3ForConditionalGeneration,image-text-to-text