transformers
8a4857c0 - Support save/load ckpt for XLA FSDP (#32311)
Commit
View On GitHub
Commit
1 year ago
Support save/load ckpt for XLA FSDP (#32311) * Support save/load ckpt for XLA FSDP * Fix bug for save * Fix style * reserve sharded ckpt and better file naming * minor fix Co-authored-by: Zach Mueller <muellerzr@gmail.com> * add is_fsdp_xla_v1_enabled --------- Co-authored-by: Zach Mueller <muellerzr@gmail.com>
Author
yitongh yitongh
Parents
f1b720ed
FAQ Terms Privacy Refunds Impressum
Loading