fix_train_bug:adapter_name
This commit is contained in:
1 parent
2a283c7e78
commit
a5be43bbbe
1 file changed
+1
-1
+1
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user