doc

Author: xadupre

_doc/conf.py

@@ -138,6 +138,7 @@ def linkcode_resolve(domain, info):
     ("py:class", "transformers.cache_utils.SlidingWindowCache"),
     ("py:class", "transformers.configuration_utils.PretrainedConfig"),
     ("py:class", "transformers.modeling_outputs.BaseModelOutput"),
+    ("py:class", "transformers.models.phi3.modeling_phi3.Phi3RotaryEmbedding"),
     ("py:func", "torch.export._draft_export.draft_export"),
     ("py:func", "torch._export.tools.report_exportability"),
     ("py:func", "torch.utils._pytree.register_pytree_node"),

onnx_diagnostic/torch_export_patches/onnx_export_errors.py

@@ -10,10 +10,8 @@
 from .patches import patch_transformers as patch_transformers_list
 
 
-def get_function(name: str) -> Tuple["module", "function"]:  # noqa: F821
-    """
-    Returns the module and the function based on its name.
-    """
+def get_function(name: str) -> Tuple[type, Callable]:
+    """Returns the module and the function based on its name."""
     spl = name.split(".")
     module_name = ".".join(spl[:-1])
     fname = spl[-1]

onnx_diagnostic/torch_export_patches/patches/patch_transformers.py

@@ -546,6 +546,7 @@ def patched__compute_dynamic_ntk_parameters(
 
     Computes the inverse frequencies with NTK scaling.
     Credits to the Reddit users /u/bloc97 and /u/emozilla
+
     Args:
         config ([`~transformers.PretrainedConfig`]):
             The model configuration.
@@ -557,6 +558,7 @@ def patched__compute_dynamic_ntk_parameters(
         rope_kwargs (`Dict`, *optional*):
             BC compatibility with the previous
             RoPE class instantiation, will be removed in v4.45.
+
     Returns:
         Tuple of (`torch.Tensor`, `float`),
         containing the inverse frequencies for the RoPE embeddings and the