accelerate
[FSDP2] Fix casting in `_cast_and_contiguous`
#3559
Merged

[FSDP2] Fix casting in `_cast_and_contiguous` #3559

S1ro1 merged 1 commit into huggingface:main from dlvp:fix-issue-3553
dlvp
cast if dtype is not None
524e20fd
SunMarc
SunMarc approved these changes on 2025-05-12
SunMarc SunMarc requested a review from S1ro1 S1ro1 326 days ago
HuggingFaceDocBuilderDev
S1ro1
S1ro1 approved these changes on 2025-05-12
S1ro1 S1ro1 merged ae0499ea into main 326 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone