transformers
8d50fda6 - Remove FSDP wrapping from sub-models. (#34452)

Commit
1 year ago
Remove FSDP wrapping from sub-models. (#34452) * Remove FSDP wrapping from sub-models. * solve conflict trainer.py * make fixup * add unit test for fsdp_auto_wrap_policy when using auto_find_batch_size * put back extract_model_from_parallel * use transformers unwrap_model
Author
Parents
Loading