From 7804177af93403d14ac8e773efe60aaf5f2d790d Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Wed, 4 Jan 2023 14:00:45 -0500 Subject: [PATCH] Fix repo consistency --- src/transformers/models/mt5/modeling_mt5.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/mt5/modeling_mt5.py b/src/transformers/models/mt5/modeling_mt5.py index ec43a03fbcf..e4e41961d5a 100644 --- 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