transformers

fix ds z3 checkpointing when `stage3_gather_16bit_weights_on_model_save=False`

#25817

Merged