Error in saving model after stage1 tuning

[applekeyword]

After stage1 tuning, use safe_save_model_for_hf_trainer to save model params, but with “AttributeError: 'Trainer' object has no attribute 'deepspeed'” error when fixing this, meet same probs: "AttributeError: 'Trainer' object has no attribute 'deepspeed'" and "AttributeError: 'Trainer' object has no attribute '_save'"