mrleo1nid/ACE-Step
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix_train_bug:adapter_name

Browse Source
This commit is contained in:
chuxij committed 2025-05-15 14:40:21 +00:00
1 parent 2a283c7e78
commit a5be43bbbe
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
trainer.py
+1 -1
View File
@@ -78,7 +78,7 @@ class Pipeline(LightningModule):
with open(lora_config_path, encoding="utf-8") as f: with open(lora_config_path, encoding="utf-8") as f:
lora_config = json.load(f) lora_config = json.load(f)
lora_config = LoraConfig(**lora_config) 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.adapter_name = adapter_name
self.transformers = transformers self.transformers = transformers