riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 11:11:05 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix T5 docstring (#20957)

Browse Source 
Fix start_docstring for deparallelize method
This commit is contained in:
ivanllt 2023-01-03 18:53:33 +08:00 committed by GitHub
parent 588faad106
commit a3e8d3cb1c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/t5/modeling_t5.py
View File

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