diff --git a/finetune.py b/finetune.py
index 64f1424..4637b95 100644
--- a/finetune.py
+++ b/finetune.py
@@ -129,7 +129,7 @@ def main(args):
     trainer.train()
 
     print("Saving the last checkpoint of the model")
-    model.save_pretrained(os.path.join(args.output_dir, "final_checkpoint/"))
+    trainer.save_model(os.path.join(args.output_dir, "final_checkpoint/"))
     if args.push_to_hub:
         trainer.push_to_hub("Upload model")
     print("Training Done! 💥")