fix ds z3 checkpointing when `stage3_gather_16bit_weights_on_model_save=False` (#25817)
* fix ds z3 checkpointing when `stage3_gather_16bit_weights_on_model_save=False` * refactoring
Showing
+14 -6
Please register or sign in to comment
* fix ds z3 checkpointing when `stage3_gather_16bit_weights_on_model_save=False` * refactoring