Update optimum/habana/transformers/models/mt5/modeling_mt5.py

Co-authored-by: Yaser Afshar <[email protected]>
Gaurav7888 · Aug 5, 2024 · 87363cb · 87363cb
1 parent 486eb63
commit 87363cb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/habana/transformers/models/mt5/modeling_mt5.py b/optimum/habana/transformers/models/mt5/modeling_mt5.py
@@ -76,8 +76,8 @@ def gaudi_MT5Attention_forward(
         use_cache=False,
         output_attentions=False,
         token_idx=None,
-    ):
-        """
+):
+    """
         Copied from https://github.com/huggingface/transformers/blob/v4.40.2/src/transformers/models/mt5/modeling_mt5.py
         Self-attention (if key_value_states is None) or attention over source sentence (provided by key_value_states).
         The only differences are: