tensor parallel dataloder for deepspeed accelerator #3390
ds tp change
d8de9fa9
update
9d1b6593
format
b0412fbf
add version check
e1160481
muellerzr
approved these changes
on 2025-03-06
Update src/accelerate/accelerator.py
b137697e
Update src/accelerate/accelerator.py
ce8be9e0
put device_mesh logic to func + format
91a723b2
fix comments
f0faeeb4
SunMarc
approved these changes
on 2025-03-11
format
8ff21019
SunMarc
merged
71036329
into main 308 days ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub