Fix lora save path

2025-02-19 10:38:26 +08:00 · 2025-02-19 10:38:26 +08:00 · 45a4bb3a0f
parent 68762f084a
commit 45a4bb3a0f
1 changed files with 1 additions and 1 deletions
--- 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()