From 6f1f26f426861e402ddcf67ff23a26b66f62e877 Mon Sep 17 00:00:00 2001 From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com> Date: Wed, 5 Apr 2023 01:30:07 +0530 Subject: [PATCH] =?UTF-8?q?=F0=9F=98=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- 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 53cbba7..74ed008 100644 --- a/src/peft/tuners/lora.py +++ b/src/peft/tuners/lora.py @@ -319,7 +319,7 @@ class LoraModel(torch.nn.Module): key_list = [key for key, _ in self.model.named_modules() if "lora" not in key] for key in key_list: - parent, target, target_name = _get_submodules(key) + parent, target, target_name = _get_submodules(self.model, key) if isinstance(target, LoraLayer): bias = target.bias is not None new_module = torch.nn.Linear(target.in_features, target.out_features, bias=bias)