style

huggingface · Oct 18, 2023 · 3d45e05 · 3d45e05
1 parent 54aa31e
commit 3d45e05
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/optimum/exporters/onnx/model_patcher.py b/optimum/exporters/onnx/model_patcher.py
@@ -464,11 +464,6 @@ def patched_forward(*args, **kwargs):
 
         self.patched_forward = patched_forward
 
-    def __exit__(self, exc_type, exc_value, traceback):
-        self.restore_ops()
-        transformers.models.falcon.modeling_falcon._make_causal_mask = self.original_make_causal
-        setattr(self._model, self.orig_forward_name, self.orig_forward)
-
 
 class WavLMModelPatcher(ModelPatcher):
     def __init__(