Fix unmask_unattended_patched signature (#1963)

* fix * style
huggingface · Jul 17, 2024 · ee07257 · ee07257
1 parent d9bd7c3
commit ee07257
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/optimum/exporters/onnx/model_patcher.py b/optimum/exporters/onnx/model_patcher.py
@@ -277,7 +277,8 @@ def __init__(
 
 
 def _unmask_unattended_patched(
-    expanded_mask: torch.Tensor, attention_mask: torch.Tensor, unmasked_value: Union[bool, float]
+    expanded_mask: torch.Tensor,
+    min_dtype: float,
 ):
     return expanded_mask