File tree Expand file tree Collapse file tree 1 file changed +3
-7
lines changed
Expand file tree Collapse file tree 1 file changed +3
-7
lines changed Original file line number Diff line number Diff line change @@ -203,14 +203,10 @@ def _get_num_processes(accelerator: str, devices: str) -> int:
203203 elif torch .backends .mps .is_available () and torch .backends .mps .is_built ():
204204 parsed_devices = MPSAccelerator .parse_devices (devices )
205205 else :
206- parsed_devices = CPUAccelerator .parse_devices (devices )
206+ return CPUAccelerator .parse_devices (devices )
207207 else :
208- parsed_devices = CPUAccelerator .parse_devices (devices )
209- return (
210- len (parsed_devices )
211- if isinstance (parsed_devices , list )
212- else (parsed_devices if isinstance (parsed_devices , int ) else 0 )
213- )
208+ return CPUAccelerator .parse_devices (devices )
209+ return len (parsed_devices ) if parsed_devices is not None else 0
214210
215211
216212def _torchrun_launch (args : Namespace , script_args : list [str ]) -> None :
You can’t perform that action at this time.
0 commit comments