complete

Author: xadupre

_doc/api/index.rst

@@ -10,6 +10,7 @@ API of onnx_diagnostic
     export/index
     helpers/index
     reference/index
+    tasks/index
     torch_export_patches/index
     torch_models/index
     torch_onnx/index

_doc/api/tasks/automatic_speech_recognition.rst

@@ -1,7 +1,7 @@
 
-onnx_diagnostic.export.automatic_speech_generation
-==================================================
+onnx_diagnostic.export.automatic_speech_recognition
+===================================================
 
-.. automodule:: onnx_diagnostic.tasks.automatic_speech_generation
+.. automodule:: onnx_diagnostic.tasks.automatic_speech_recognition
     :members:
     :no-undoc-members:

_doc/api/tasks/index.rst

@@ -5,13 +5,13 @@ onnx_diagnostic.tasks
     :maxdepth: 1
     :caption: modules
 
-    automatic_speech_generation
+    automatic_speech_recognition
     fill_mask
     image_classification
     image_text_to_text      
     text_generation
     text2text_generation
-    zero_short_image_classification
+    zero_shot_image_classification
 
 reduce_model_config
 +++++++++++++++++++

_doc/api/tasks/zero_shot_image_classification.rst

@@ -1,7 +1,7 @@
 
-onnx_diagnostic.export.zero_short_image_classification
-======================================================
+onnx_diagnostic.export.zero_shot_image_classification
+=====================================================
 
-.. automodule:: onnx_diagnostic.tasks.zero_short_image_classification
+.. automodule:: onnx_diagnostic.tasks.zero_shot_image_classification
     :members:
     :no-undoc-members:

_unittests/ut_tasks/test_tasks.py

@@ -96,7 +96,7 @@ def test_fill_mask(self):
         mid = "google-bert/bert-base-multilingual-cased"
         # mid = "Salesforce/codet5-small"
         data = get_untrained_model_with_inputs(mid, verbose=1)
-        self.assertIn((data["size"], data["n_weights"]), [(12742888, 3185722)])
+        self.assertIn((data["size"], data["n_weights"]), [(428383212, 107095803)])
         model, inputs = data["model"], data["inputs"]
         model(**inputs)
 

_unittests/ut_torch_models/test_hghub_api.py

@@ -100,7 +100,7 @@ def test_hf_all_models(self):
 
     def test_load_architecture_task(self):
         data = load_architecture_task()
-        print(set(data.values()))
+        self.assertNotEmpty(set(data.values()))
 
     def test_task_from_tags(self):
         _tags = [
@@ -132,10 +132,10 @@ def test_model_testings_and_architectures(self):
 
     def test__ccached_config_64(self):
         from onnx_diagnostic.torch_models.hghub.hub_data_cached_configs import (
-            _cached_hf_internal_testing_tiny_random_beitforimageclassification,
+            _ccached_hf_internal_testing_tiny_random_beitforimageclassification,
         )
 
-        conf = _cached_hf_internal_testing_tiny_random_beitforimageclassification()
+        conf = _ccached_hf_internal_testing_tiny_random_beitforimageclassification()
         self.assertEqual(conf.auxiliary_channels, 256)
 
 

onnx_diagnostic/tasks/zero_shot_image_classification.py

@@ -2,7 +2,7 @@
 import torch
 from ..helpers.config_helper import update_config, check_hasattr
 
-__TASK__ = "zero-shot-image-generation"
+__TASK__ = "zero-shot-image-classification"
 
 
 def reduce_model_config(config: Any, task: str) -> Dict[str, Any]: