From 59e65001054f47a413a2d19ae8bd816a2f25d3ae Mon Sep 17 00:00:00 2001
From: fxmarty <9808326+fxmarty@users.noreply.github.com>
Date: Wed, 17 Jul 2024 14:41:53 +0200
Subject: [PATCH] fix

---
 optimum/exporters/onnx/model_patcher.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/optimum/exporters/onnx/model_patcher.py b/optimum/exporters/onnx/model_patcher.py
index 5e720d0cd7..09d3ab2268 100644
--- a/optimum/exporters/onnx/model_patcher.py
+++ b/optimum/exporters/onnx/model_patcher.py
@@ -277,7 +277,7 @@ 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