diff --git a/CHANGELOGS.rst b/CHANGELOGS.rst index ce0b259b..b6d36fa3 100644 --- a/CHANGELOGS.rst +++ b/CHANGELOGS.rst @@ -1,6 +1,9 @@ Change Logs =========== +0.4.3 ++++++ + 0.4.2 +++++ diff --git a/_doc/index.rst b/_doc/index.rst index d6fc9c9a..917df6b1 100644 --- a/_doc/index.rst +++ b/_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>`_ diff --git a/onnx_diagnostic/__init__.py b/onnx_diagnostic/__init__.py index 68fe5b78..2d02cd2f 100644 --- a/onnx_diagnostic/__init__.py +++ b/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é" diff --git a/onnx_diagnostic/torch_models/hghub/hub_data.py b/onnx_diagnostic/torch_models/hghub/hub_data.py index 5ec2abf2..ac4565f7 100644 --- a/onnx_diagnostic/torch_models/hghub/hub_data.py +++ b/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""" )