Remove FSDP wrapping from sub-models.#34452

Merged

ArthurZucker merged 6 commits intohuggingface:mainfrom

eljandoubi:fix_fsdp_auto_wrap_policy

Nov 15, 2024

Commits on Nov 5, 2024

Remove FSDP wrapping from sub-models.
eljandoubi
committed
solve conflict trainer.py

eljandoubi
authored and
eljandoubi
committed
make fixup
eljandoubi
committed
add unit test for fsdp_auto_wrap_policy when using auto_find_batch_size
eljandoubi
committed
put back extract_model_from_parallel

eljandoubi
authored and
eljandoubi
committed
use transformers unwrap_model

eljandoubi
authored and
eljandoubi
committed