hiyouga 90f706c05b fix ppo trainer save zero3 model
accelerator.get_state_dict(ds_model) should be called at all ranks


Former-commit-id: 3a0f60f0aa072531e4ae5819ec00c8fa42aa0913
2024-06-07 05:14:19 +08:00
..
2024-05-16 18:39:08 +08:00
2024-06-07 05:14:19 +08:00
2024-06-07 00:09:06 +08:00
2024-06-07 00:09:06 +08:00