Update utils.py

2025-02-02 16:26:55 -08:00 · 2025-02-02 16:26:55 -08:00 · 8a6f9d4f05
commit 8a6f9d4f05
parent 379356c5f2
1 changed files with 1 additions and 0 deletions
--- a/unsloth/kernels/utils.py
+++ b/unsloth/kernels/utils.py
@ -219,6 +219,7 @@ else:

            # Use same buffers for faster inference
            size = shape[0]*shape[1]
+            print(shape, size)
            global WEIGHT_BUFFER
            global ABSMAX_BUFFER
            if WEIGHT_BUFFER is None: