Merge pull request #187 from ace-step/fix_save_lora_name

fix_train_bug:adapter_name
2025-05-15 22:41:56 +08:00
parent ff3dc5a8ce a5be43bbbe
commit 6c14becf98
1 changed files with 1 additions and 1 deletions
@@ -78,7 +78,7 @@ class Pipeline(LightningModule):
            with open(lora_config_path, encoding="utf-8") as f:
                lora_config = json.load(f)
            lora_config = LoraConfig(**lora_config)
-            transformers.add_adapter(adapter_config=lora_config)
+            transformers.add_adapter(adapter_config=lora_config, adapter_name=adapter_name)
            self.adapter_name = adapter_name

        self.transformers = transformers