From f41a437540ee60e00d27147c49a0b51a9de37c71 Mon Sep 17 00:00:00 2001
From: Daniel Han-Chen <danielhanchen@gmail.com>
Date: Mon, 22 Jan 2024 22:34:33 +1100
Subject: [PATCH] Update utils.py

---
 unsloth/kernels/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/unsloth/kernels/utils.py b/unsloth/kernels/utils.py
index 5f5e7ac5e7..26d0f6bb4d 100644
--- a/unsloth/kernels/utils.py
+++ b/unsloth/kernels/utils.py
@@ -180,7 +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(X.shape, W.shape, out.shape)
+    print(X.shape, W.quant_state.shape, out.shape)
     if lora_A is not None:
         dtype = X.dtype
         temp_lora = torch.matmul(X, lora_A.to(dtype).t(), out = temp_lora)