π Describe the bug
error indent:

correct indent:

error when save model
β /home/bmb/projects/ColossalAI/applications/Chat/examples/train_prompts.py:168 in main β
β β
β 165 β β β β update_timesteps=args.update_timesteps) β
β 166 β β
β 167 β # save model checkpoint after fitting β
β β± 168 β trainer.save_model(args.save_path, only_rank0=True, tokenizer=tokenizer) β
β 169 β # save optimizer checkpoint on all ranks β
β 170 β if args.need_optim_ckpt: β
β 171 β β strategy.save_optimizer(actor_optim, β
β°βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ―
AttributeError: 'PPOTrainer' object has no attribute 'save_model'
Environment
No response
π Describe the bug
error indent:


correct indent:
error when save model
β /home/bmb/projects/ColossalAI/applications/Chat/examples/train_prompts.py:168 in main β
β β
β 165 β β β β update_timesteps=args.update_timesteps) β
β 166 β β
β 167 β # save model checkpoint after fitting β
β β± 168 β trainer.save_model(args.save_path, only_rank0=True, tokenizer=tokenizer) β
β 169 β # save optimizer checkpoint on all ranks β
β 170 β if args.need_optim_ckpt: β
β 171 β β strategy.save_optimizer(actor_optim, β
β°βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ―
AttributeError: 'PPOTrainer' object has no attribute 'save_model'
Environment
No response