transformers
Fix mixed torch.Tensor and DTensor in generate when use FSDP2 + LoRA
#42436

Merged

Fix mixed torch.Tensor and DTensor in generate when use FSDP2 + LoRA #42436

3outeille merged 3 commits into huggingface:main from Xiao-Chenguang:main

3outeille requested changes on 2025-12-02

Fix mixed torch.Tensor and DTensor error by registering fsdp forward …

9cabb1cd

Apply fsdp forward register when fsdp is enabled

ee9152e7

Xiao-Chenguang force pushed from 7a0c9669 to ee9152e7 342 days ago

Merge branch 'main' into main

bf3534e2

3outeille approved these changes on 2025-12-03

3outeille merged 17c7c496 into main 342 days ago

Reviewers

3outeille

Assignees

No one assigned

Labels

None yet

Milestone

No milestone