Skip to content

fix: add PyTorch version check for torch.accelerator in gather_size_by_comm#13093

Open
Mr-Neutr0n wants to merge 2 commits into
huggingface:mainfrom
Mr-Neutr0n:fix/pytorch-version-accelerator-check
Open

fix: add PyTorch version check for torch.accelerator in gather_size_by_comm#13093
Mr-Neutr0n wants to merge 2 commits into
huggingface:mainfrom
Mr-Neutr0n:fix/pytorch-version-accelerator-check

Simplify gather_device selection per @DN6 suggestion

3a12e93
Select commit
Loading
Failed to load commit list.
Sign in for the full log view