Update lora.py

This commit is contained in:
Sourab Mangrulkar
2023-02-17 17:48:16 +05:30
parent 173dc3dedf
commit 1a8928c5a4
+3 -1
View File
@@ -494,6 +494,8 @@ if is_bnb_available():
def forward(self, x: torch.Tensor):
result = super().forward(x)
if self.r > 0:
if self.disable_adapters:
return result
elif self.r > 0:
result += self.lora_B(self.lora_A(self.lora_dropout(x))) * self.scaling
return result