SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Muellerzr fix deepspeed (#23657)

Browse Source 
* Fix deepspeed recursion

* Better fix
This commit is contained in:
Zachary Mueller
2023-05-22 11:22:54 -04:00
committed by GitHub
parent 7bbdfd7b24
commit fe34486f12
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/training_args.py
View File

@@ -1708,6 +1708,7 @@ class TrainingArguments:
""" """
requires_backends(self, ["torch"]) requires_backends(self, ["torch"])
# Make sure `self._n_gpu` is properly setup. # Make sure `self._n_gpu` is properly setup.
if not hasattr(self, "_n_gpu"):
_ = self._setup_devices _ = self._setup_devices
return self._n_gpu return self._n_gpu