from local

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

View File

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