fix gpt bigcode ONNX export for transformers<4.39.0

huggingface · Jul 25, 2024 · 2997f94 · 2997f94
1 parent 21f709c
commit 2997f94
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/optimum/exporters/onnx/model_patcher.py b/optimum/exporters/onnx/model_patcher.py
@@ -281,9 +281,11 @@ def _unmask_unattended_patched_legacy(
 ):
     return expanded_mask
 
+
 def _unmask_unattended_patched(expanded_mask: torch.Tensor, min_dtype: float):
     return expanded_mask
 
+
 def _make_causal_mask_patched(
     input_ids_shape: torch.Size,
     dtype: torch.dtype,