Add more architectures (#74)

Author: xadupre

CHANGELOGS.rst

@@ -1,6 +1,9 @@
 Change Logs
 ===========
 
+0.4.3
++++++
+
 0.4.2
 +++++
 

_doc/index.rst

@@ -173,6 +173,7 @@ Size of the package:
 Older versions
 ++++++++++++++
 
+* `0.4.3 <../v0.4.3/index.html>`_
 * `0.4.2 <../v0.4.2/index.html>`_
 * `0.4.1 <../v0.4.1/index.html>`_
 * `0.4.0 <../v0.4.0/index.html>`_

onnx_diagnostic/__init__.py

@@ -3,5 +3,5 @@
 Functions, classes to dig into a model when this one is right, slow, wrong...
 """
 
-__version__ = "0.4.2"
+__version__ = "0.4.3"
 __author__ = "Xavier Dupré"

onnx_diagnostic/torch_models/hghub/hub_data.py

@@ -38,12 +38,14 @@
     DeiTModel,image-feature-extraction
     DetrModel,image-feature-extraction
     Dinov2Model,image-feature-extraction
+    DistilBertForSequenceClassification,text-classification
     DistilBertModel,feature-extraction
     DonutSwinModel,feature-extraction
     ElectraModel,feature-extraction
     EsmModel,feature-extraction
     FalconMambaForCausalLM,text-generation
     GLPNModel,image-feature-extraction
+    GPT2LMHeadModel,text-generation
     GPTBigCodeModel,feature-extraction
     GPTJModel,feature-extraction
     GPTNeoModel,feature-extraction
@@ -64,6 +66,7 @@
     LongT5Model,feature-extraction
     LongformerModel,feature-extraction
     MCTCTModel,feature-extraction
+    MPNetForMaskedLM,sentence-similarity
     MPNetModel,feature-extraction
     MT5Model,feature-extraction
     MarianMTModel,text2text-generation
@@ -96,11 +99,13 @@
     PoolFormerModel,image-feature-extraction
     PvtForImageClassification,image-classification
     Qwen2ForCausalLM,text-generation
+    Qwen2_5_VLForConditionalGeneration,image-text-to-text
     RTDetrForObjectDetection,object-detection
     RegNetModel,image-feature-extraction
     RemBertModel,feature-extraction
     ResNetForImageClassification,image-classification
     RoFormerModel,feature-extraction
+    RobertaForMaskedLM,sentence-similarity
     RobertaModel,feature-extraction
     RtDetrV2ForObjectDetection,object-detection
     SEWDModel,feature-extraction
@@ -118,6 +123,7 @@
     Swinv2Model,image-feature-extraction
     T5ForConditionalGeneration,text2text-generation
     TableTransformerModel,image-feature-extraction
+    TableTransformerForObjectDetection,object-detection
     UniSpeechForSequenceClassification,audio-classification
     ViTForImageClassification,image-classification
     ViTMAEModel,image-feature-extraction
@@ -130,6 +136,9 @@
     WhisperForConditionalGeneration,automatic-speech-recognition
     XLMModel,feature-extraction
     XLMRobertaForCausalLM,text-generation
+    XLMRobertaForMaskedLM,fill-mask
+    XLMRobertaModel,sentence-similarity
+    Wav2Vec2ForCTC,automatic-speech-recognition
     YolosForObjectDetection,object-detection
     YolosModel,image-feature-extraction"""
 )