From 6684a6364d3f5f762f07ebe103d6d4b7fabf77f4 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Thu, 25 Jul 2024 16:49:18 +0200 Subject: [PATCH] format --- optimum/exporters/onnx/model_patcher.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/optimum/exporters/onnx/model_patcher.py b/optimum/exporters/onnx/model_patcher.py index af3c222e866..4c1f8458930 100644 --- 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,