wassname/peft
1
0
Fork 0
mirror of https://github.com/wassname/peft.git synced 2026-09-21 13:10:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update lora.py

Browse Source
This commit is contained in:
Sourab Mangrulkar
2023-03-28 19:34:04 +05:30
parent 7d7c598647
commit 64cae2aab2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/peft/tuners/lora.py
+1 -1
View File
@@ -139,7 +139,7 @@ class LoraModel(torch.nn.Module):
if target_module_found:
if not is_target_modules_in_base_model:
is_target_modules_in_base_model = True
parent, target, target_name = _get_submodules(key)
parent, target, target_name = _get_submodules(self.model, key)
bias = target.bias is not None
if isinstance(target, LoraLayer):
target.update_layer(adapter_name, lora_config.r, lora_config.lora_alpha, lora_config.lora_dropout)