wassname/peft
1
0
Fork 0
mirror of https://github.com/wassname/peft.git synced 2026-09-10 12:20:21 +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:36:41 +05:30
parent e9d45da4c5
commit 8ec7cb8435
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
@@ -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):