transformers
031ef880 - fix FSDP + torch.compile bug when saving pretrained model (#37725)

Commit
302 days ago
fix FSDP + torch.compile bug when saving pretrained model (#37725) * args keep_torch_compile=False in _save and _wwrap_method * Fix FSDP execution on evaluation for torch_compile mode * add test trainer FSDP + Torch Compile * fix quality code * make style * Revert " make style" This reverts commit 77e797f8829c50992cc21496be3d9a3e480e1c97. * make style
Author
Parents
Loading