save_model -> save_pretrained in ppo_trainer.mdx (huggingface#1537)

alexvishnevskiy · Apr 15, 2024 · e823458 · e823458
1 parent 1c0d8bc
commit e823458
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/ppo_trainer.mdx b/docs/source/ppo_trainer.mdx
@@ -151,7 +151,7 @@ for epoch in tqdm(range(epochs), "epoch: "):
         ppo_trainer.log_stats(stats, batch, rewards)
 
 #### Save model
-ppo_trainer.save_model("my_ppo_model")
+ppo_trainer.save_pretrained("my_ppo_model")
 ```
 
 ## Logging