from local

This commit is contained in:
dohe0342 2023-05-18 17:09:42 +09:00
parent 6d8ceec54e
commit 52fbf84ffa
2 changed files with 1 additions and 1 deletions

View File

@ -267,7 +267,7 @@ class LoRAModule(nn.Module):
x = x.transpose(0, 1)
x = self.lora_A[layer_idx](x)
x = self.lora_B[layer_idx](x)
x = x.transpose(0, 1)
'''