From 64cae2aab2174214987758f0486c3d6ae8a96563 Mon Sep 17 00:00:00 2001 From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com> Date: Tue, 28 Mar 2023 19:34:04 +0530 Subject: [PATCH] Update lora.py --- src/peft/tuners/lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/peft/tuners/lora.py b/src/peft/tuners/lora.py index 34cee7d..5459381 100644 --- a/src/peft/tuners/lora.py +++ b/src/peft/tuners/lora.py @@ -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)