From cb8d4fc5a40c8d688e23e004258287e1ed5fba27 Mon Sep 17 00:00:00 2001
From: qianyu chen <38046403+qyc-98@users.noreply.github.com>
Date: Wed, 17 Jul 2024 09:30:35 +0800
Subject: [PATCH] Update finetune.py

---
 finetune/finetune.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/finetune/finetune.py b/finetune/finetune.py
index 671d6c7..d2d2cff 100644
--- a/finetune/finetune.py
+++ b/finetune/finetune.py
@@ -260,6 +260,7 @@ def train():
             lora_dropout=lora_args.lora_dropout,
             bias=lora_args.lora_bias,
             layers_to_transform=lora_args.lora_layers_to_transform,
+            modules_to_save=modules_to_save,
         )
         if not hasattr(model, 'get_input_embeddings'):
             def get_input_embeddings(self):