doc

Author: xadupre

_doc/conf.py

@@ -121,6 +121,7 @@
     ("py:class", "transformers.LlamaConfig"),
     ("py:class", "transformers.cache_utils.Cache"),
     ("py:class", "transformers.cache_utils.DynamicCache"),
+    ("py:class", "transformers.cache_utils.EncoderDecoderCache"),
     ("py:class", "transformers.cache_utils.MambaCache"),
     ("py:func", "torch.export._draft_export.draft_export"),
     ("py:func", "torch._export.tools.report_exportability"),

onnx_diagnostic/torch_export_patches/patch_inputs.py

@@ -147,7 +147,7 @@ def convert_dynamic_axes_into_dynamic_shapes(
                 # dynamic axes not in the given inputs, should be raise an exception?
                 if verbose:
                     print(
-                        f"[convert_dynamic_axes_into_dynamic_shapes] droping axes "
+                        f"[convert_dynamic_axes_into_dynamic_shapes] dropping axes "
                         f"{k!r}-{v!r}, not found in {set(updated_kwargs)}"
                     )
                 continue