Commit Graph
3 Commits
Author SHA1 Message Date
hiyouga 4489d73ac7 fix ppo trainer save zero3 model
accelerator.get_state_dict(ds_model) should be called at all ranks
2024-06-07 05:14:19 +08:00
hiyouga 74f96efef9 rename files 2024-06-07 00:09:06 +08:00
hiyouga 308edbc426 rename package 2024-05-16 18:39:08 +08:00