fix format

huggingface · Sep 20, 2023 · 26d97e8 · 26d97e8
1 parent b05f599
commit 26d97e8
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/optimum/exporters/onnx/model_patcher.py b/optimum/exporters/onnx/model_patcher.py
@@ -402,7 +402,11 @@ def __init__(
         model_kwargs: Optional[Dict[str, Any]] = None,
     ):
         super().__init__(config, model, model_kwargs)
-        self.patch = self.real_config.task == "text-generation" and self.real_config.use_past and self.real_config._behavior == "decoder"
+        self.patch = (
+            self.real_config.task == "text-generation"
+            and self.real_config.use_past
+            and self.real_config._behavior == "decoder"
+        )
         if self.patch:
             self.orig_prepare_attn_mask = getattr(self._model.model.decoder, "_prepare_decoder_attention_mask")
 
@@ -440,7 +444,6 @@ def __exit__(self, exc_type, exc_value, traceback):
             setattr(self._model.model.decoder, "_prepare_decoder_attention_mask", self.orig_prepare_attn_mask)
 
 
-
 class MPTModelPatcher(BloomModelPatcher):
     pass
 
@@ -455,4 +458,3 @@ class BlenderbotModelPatcher(BartModelPatcher):
 
 class PegasusModelPatcher(BartModelPatcher):
     pass
-