Fix repo consistency

7804177a · Sylvain Gugger · 15e17c99 · 7804177a
Unverified Commit 7804177a authored Jan 04, 2023 by Sylvain Gugger
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/mt5/modeling_mt5.py src/transformers/models/mt5/modeling_mt5.py +1 -1

No files found.
--- a/src/transformers/models/mt5/modeling_mt5.py
+++ b/src/transformers/models/mt5/modeling_mt5.py
@@ -862,7 +862,7 @@ class MT5Stack(MT5PreTrainedModel):
        # Set final layer norm to last device
        self.final_layer_norm = self.final_layer_norm.to(self.last_device)

-    @add_start_docstrings(PARALLELIZE_DOCSTRING)
+    @add_start_docstrings(DEPARALLELIZE_DOCSTRING)
    def deparallelize(self):
        self.model_parallel = False
        self.device_map = None