From 45a4bb3a0fc3fa7517f9da560f37b1d09e4b632a Mon Sep 17 00:00:00 2001
From: xunuohope1107 <148425765+xunuohope1107@users.noreply.github.com>
Date: Wed, 19 Feb 2025 10:38:26 +0800
Subject: [PATCH] Fix lora save path

---
 train_lora.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train_lora.py b/train_lora.py
index c4877d9..db5841c 100644
--- a/train_lora.py
+++ b/train_lora.py
@@ -80,7 +80,7 @@ def train_epoch(epoch, wandb):
         if (step + 1) % args.save_interval == 0 and (not ddp or dist.get_rank() == 0):
             model.eval()
             # 【区别1】只保存lora权重即可
-            save_lora(model, f'{args.save_dir}/lora/{args.lora_name}_{lm_config.dim}.pth')
+            save_lora(model, f'{args.save_dir}/{args.lora_name}_{lm_config.dim}.pth')
             model.train()