From 71a1f7f38c400b0feb9efcd4b30187250ab007d4 Mon Sep 17 00:00:00 2001
From: NanoCode012 <kevinvong@rocketmail.com>
Date: Wed, 10 May 2023 01:08:22 +0900
Subject: [PATCH] Save adapter for lora

---
 scripts/finetune.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/scripts/finetune.py b/scripts/finetune.py
index 1da332f40..a7c3d6c77 100644
--- a/scripts/finetune.py
+++ b/scripts/finetune.py
@@ -230,6 +230,9 @@ def train(
     )
     # TODO do we need this fix? https://huggingface.co/docs/accelerate/usage_guides/fsdp#saving-and-loading
     trainer.save_model(cfg.output_dir)
+    
+    if cfg.adapter == 'lora':
+        trainer.save_pretrained(cfg.output_dir)
 
 
 if __name__ == "__main__":