From 0628529372d5cfb5e1f40391185c1f498eacee71 Mon Sep 17 00:00:00 2001 From: Ti-Tai Wang Date: Mon, 25 Aug 2025 21:50:02 +0000 Subject: [PATCH] add new arch --- onnx_diagnostic/torch_models/hghub/hub_data.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/onnx_diagnostic/torch_models/hghub/hub_data.py b/onnx_diagnostic/torch_models/hghub/hub_data.py index 517f434f..0b752ec1 100644 --- a/onnx_diagnostic/torch_models/hghub/hub_data.py +++ b/onnx_diagnostic/torch_models/hghub/hub_data.py @@ -37,6 +37,7 @@ DebertaModel,feature-extraction DebertaV2Model,feature-extraction DecisionTransformerModel,reinforcement-learning + DeepseekV3ForCausalLM,text-generation DeiTModel,image-feature-extraction DetrModel,image-feature-extraction Dinov2Model,image-feature-extraction @@ -56,6 +57,8 @@ GemmaForCausalLM,text-generation Gemma2ForCausalLM,text-generation Gemma3ForConditionalGeneration,image-text-to-text + Gemma3ForCausalLM,text-generation + Glm4vMoeForConditionalGeneration,image-text-to-text GraniteForCausalLM,text-generation GroupViTModel,feature-extraction HieraForImageClassification,image-classification @@ -108,6 +111,7 @@ PvtForImageClassification,image-classification Qwen2ForCausalLM,text-generation Qwen2_5_VLForConditionalGeneration,image-text-to-text + Qwen3MoeForCausalLM,text-generation RTDetrForObjectDetection,object-detection RegNetModel,image-feature-extraction RemBertModel,feature-extraction