From 8ec7cb84350239d6ca51aa1223c91d40714f6a1b Mon Sep 17 00:00:00 2001 From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com> Date: Tue, 28 Mar 2023 19:36:41 +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 215a315..058a3f3 100644 --- a/src/peft/tuners/lora.py +++ b/src/peft/tuners/lora.py @@ -444,7 +444,7 @@ if is_bnb_available(): # Freezing the pre-trained weight matrix self.weight.requires_grad = False - self.update_layer(self, adapter_name, r, lora_alpha, lora_dropout) + self.update_layer(adapter_name, r, lora_alpha, lora_dropout) self.active_adapter = adapter_name def forward(self, x: torch.Tensor):