Update lora.py

This commit is contained in:
Sourab Mangrulkar
2023-03-28 19:35:49 +05:30
parent 64cae2aab2
commit e9d45da4c5
+1 -1
View File
@@ -361,7 +361,7 @@ class Linear(nn.Linear):
self.weight.data = self.weight.data.T
nn.Linear.reset_parameters(self)
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 merge(self):