PR #4021 Fix fsdp2 load full state dict dtype mismatch

Fix fsdp2 load full state dict dtype mismatch #4021

SunMarc merged 2 commits into main from fix-fsdp2-load-full-state-dict-dtype-mismatch

[FSDP2] Capture original_sd after upcast to fix dtype mismatch hang

b611ab10

Fix

73ae51ba

SunMarc merged d1d2bba4 into main 38 days ago

SunMarc deleted the fix-fsdp2-load-full-state-dict-dtype-mismatch branch 38 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone