diff --git a/unsloth/kernels/fast_lora.py b/unsloth/kernels/fast_lora.py index 04956fd7aa..4ecf9589bf 100644 --- a/unsloth/kernels/fast_lora.py +++ b/unsloth/kernels/fast_lora.py @@ -28,6 +28,7 @@ def matmul_lora(X, W, W_quant, A, B, s, out = None): reshape = False pass + A, B = A.t(), B.t() W.addmm_(A.to(dtype), B.to(dtype), alpha = s) out = torch.matmul(X, W, out = out)