[FSDP2] Cast model to uniform dtype before fully_shard to fix mixed-dtype AssertionError#3985
Open
roycho96 wants to merge 2 commits intohuggingface:mainfrom
Open
[FSDP2] Cast model to uniform dtype before fully_shard to fix mixed-dtype AssertionError#3985roycho96 wants to merge 2 commits intohuggingface:mainfrom
roycho96 wants to merge 2 commits intohuggingface:mainfrom
