Save adapter for lora

2023-05-10 01:08:22 +09:00
parent 02c59832a3
commit 71a1f7f38c
1 changed files with 3 additions and 0 deletions
--- 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__":