diffusers
[Context Parallel] fix: support PyTorch < 2.6 in distributed gather
#13075

Open

[Context Parallel] fix: support PyTorch < 2.6 in distributed gather #13075

akshatvishu wants to merge 3 commits into huggingface:main from akshatvishu:fix-torch-accelerator-compatibility

[Context Parallel] fix: support PyTorch < 2.6 in distributed gather

a46c3502

sayakpaul commented on 2026-02-04

[Context Parallel] fix: use get_device() and add fail-fast check

f01c13c2

sayakpaul commented on 2026-02-04

fix: use torch.device with get_device and simplify error message

b300f055

Reviewers

sayakpaul

Assignees

No one assigned

Labels

None yet

Milestone

No milestone