Fix T5 docstring (#20957)

Fix start_docstring for deparallelize method

Fix T5 docstring (#20957)
Fix start_docstring for deparallelize method
a3e8d3cb · ivanllt · GitHub · 588faad1 · a3e8d3cb
Unverified Commit a3e8d3cb authored Jan 03, 2023 by ivanllt Committed by GitHub Jan 03, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/t5/modeling_t5.py src/transformers/models/t5/modeling_t5.py +1 -1

No files found.
--- a/src/transformers/models/t5/modeling_t5.py
+++ b/src/transformers/models/t5/modeling_t5.py
@@ -891,7 +891,7 @@ class T5Stack(T5PreTrainedModel):
        # 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