"docs/vscode:/vscode.git/clone" did not exist on "1182b945a6ea0f8cf53b1058d383fb9bc5ff02e8"
Unverified Commit fe34486f authored by Zachary Mueller's avatar Zachary Mueller Committed by GitHub
Browse files

Muellerzr fix deepspeed (#23657)

* Fix deepspeed recursion

* Better fix
parent 7bbdfd7b
......@@ -1708,6 +1708,7 @@ class TrainingArguments:
"""
requires_backends(self, ["torch"])
# Make sure `self._n_gpu` is properly setup.
if not hasattr(self, "_n_gpu"):
_ = self._setup_devices
return self._n_gpu
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment