Skip to content

Fix model saving bug post training with tensor parallel in Accelerate #36434

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
bursteratom wants to merge 4 commits into from

add .cpu() when gathering tensor

ee271a0
Select commit
Loading
Failed to load commit list.
Closed

Fix model saving bug post training with tensor parallel in Accelerate #36434

add .cpu() when gathering tensor
ee271a0
Select commit
Loading
Failed to load commit list.
CircleCI Checks / setup_and_quality succeeded Jun 19, 2025 in 4m 13s

Workflow: setup_and_quality

View CircleCI Workflow

View more details on CircleCI Checks