PR #43226 fix crash in when running FSDP2+TP

fix crash in when running FSDP2+TP #43226

ArthurZucker merged 5 commits into huggingface:main from sywangyi:fsdp+tp

fix crash in when running FSDP2+TP

1c054cb3

SunMarc commented on 2026-01-12

update

d50001e0

Merge branch 'main' into fsdp+tp

4bc90a39

fix tp only issue

07454dc0

Merge branch 'main' into fsdp+tp

754498fd

SunMarc approved these changes on 2026-01-14

ArthurZucker approved these changes on 2026-01-14

ArthurZucker merged 77146cc9 into main 3 days ago

Reviewers

ArthurZucker

SunMarc

Assignees

No one assigned

Labels

None yet

Milestone

No milestone