riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Fix repo consistency

Browse Source
This commit is contained in:
Sylvain Gugger 2023-01-04 14:00:45 -05:00
parent 15e17c99f9
commit 7804177af9
No known key found for this signature in database
GPG Key ID: 4198F83D867EFC2E
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/mt5/modeling_mt5.py
View File

@ -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