Commit Graph

5 Commits

Author SHA1 Message Date
hiyouga
6baafd4eb3 fix #4221 2024-06-13 02:48:21 +08:00
hiyouga
2ed8270112 clean code 2024-06-13 01:58:16 +08:00
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