patch

Author: xadupre

onnx_diagnostic/torch_export_patches/onnx_export_errors.py

@@ -32,7 +32,7 @@ def get_patches(mod, verbose: int = 0) -> Tuple[str, List[Any]]:
                 to_patch.append(v)
             else:
                 # a function
-                doc = v.__doc__
+                doc = v.__doc__.lstrip()
                 if doc.startswith("manual patch"):
                     continue
                 reg = re.compile("[[]patch:([a-z_A-Z.]+)[]]")

onnx_diagnostic/torch_export_patches/patches/patch_transformers.py

@@ -540,7 +540,8 @@ def patched__compute_dynamic_ntk_parameters(
     seq_len: Optional[int] = None,
     **rope_kwargs,
 ) -> Tuple["torch.Tensor", float]:
-    """manual patch:
+    """
+    manual patch:
     ``[patch:transformers.modeling_rope_utils._compute_dynamic_ntk_parameters]``
 
     Computes the inverse frequencies with NTK scaling.