From 2e29e839edbe490dc03d797d180bec66835cdd0f Mon Sep 17 00:00:00 2001
From: OsamaS99 <62110783+OsamaS99@users.noreply.github.com>
Date: Thu, 21 Mar 2024 12:52:39 +0100
Subject: [PATCH] Update finetune.py

Save model and adapters.
---
 finetune.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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! 💥")