doc

Author: xadupre

_doc/api/export/index.rst

@@ -10,7 +10,7 @@ onnx_diagnostic.export
 ModelInputs
 +++++++++++
 
-.. autoclass:: onnx_diagnostic.dyanmic_shapes.ModelInputs
+.. autoclass:: onnx_diagnostic.export.ModelInputs
     :members:
 
 Other functions

_doc/conf.py

@@ -115,13 +115,13 @@
     ("py:class", "torch.utils._pytree.Context"),
     ("py:class", "torch.utils._pytree.KeyEntry"),
     ("py:class", "torch.utils._pytree.TreeSpec"),
-    ("py:class", "transformers.GenerationMixin.generate"),
     ("py:class", "transformers.LlamaConfig"),
     ("py:class", "transformers.cache_utils.Cache"),
     ("py:class", "transformers.cache_utils.DynamicCache"),
     ("py:class", "transformers.cache_utils.MambaCache"),
     ("py:func", "torch.export._draft_export.draft_export"),
     ("py:func", "torch._export.tools.report_exportability"),
+    ("py:meth", "transformers.GenerationMixin.generate"),
 ]
 
 nitpick_ignore_regex = [