doc

Author: xadupre

_doc/conf.py

@@ -124,6 +124,7 @@
     ("py:class", "transformers.cache_utils.MambaCache"),
     ("py:func", "torch.export._draft_export.draft_export"),
     ("py:func", "torch._export.tools.report_exportability"),
+    ("py:meth", "huggingface_hub.HfApi.list_models"),
     ("py:meth", "transformers.GenerationMixin.generate"),
 ]
 

onnx_diagnostic/__init__.py

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

onnx_diagnostic/torch_models/hghub/hub_api.py

@@ -25,7 +25,7 @@ def task_from_arch(arch: str) -> str:
 
     .. runpython::
 
-        from onnx_diagnostic.torch_models.hub_data import __date__
+        from onnx_diagnostic.torch_models.hghub.hub_data import __date__
         print("last refresh", __date__)
 
     List of supported architectures, see