Howard Liberty f16caf44bb Add FSDP config for CPU RAM efficient loading through accelerate (#30002) ... * Add FSDP config for CPU RAM efficient loading * Style fix * Update src/transformers/training_args.py Co-authored-by: Zach Mueller <muellerzr@gmail.com> * Update src/transformers/training_args.py Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> * Add sync_module_states and cpu_ram_efficient_loading validation logic * Update src/transformers/training_args.py Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> * Style --------- Co-authored-by: Zach Mueller <muellerzr@gmail.com> Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>		2024-04-22 13:15:28 +01:00
..
test_fsdp.py	Add FSDP config for CPU RAM efficient loading through accelerate (#30002)	2024-04-22 13:15:28 +01:00