Update finetune.py

2026-02-04 17:59:18 +08:00 · 2024-07-17 09:30:35 +08:00
parent dc68caefbd
commit cb8d4fc5a4
1 changed files with 1 additions and 0 deletions
--- 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):