diff --git a/unsloth/kernels/utils.py b/unsloth/kernels/utils.py index 141814c332..7051d062de 100644 --- a/unsloth/kernels/utils.py +++ b/unsloth/kernels/utils.py @@ -180,6 +180,7 @@ pass def fast_linear_forward(proj, X, temp_lora = None, out = None): W, W_quant, lora_A, lora_B, lora_S = get_lora_parameters(proj) out = fast_gemv(X, W, W_quant, out = out) + print(out.shape, X.shape) if lora_A is not None: dtype = X.dtype temp_lora = torch.matmul(X, lora_A.to(dtype).t(), out = temp_lora)