From 8a6f9d4f05121c1da1e391cbc3be91470ed6df52 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 2 Feb 2025 16:26:55 -0800 Subject: [PATCH] Update utils.py --- unsloth/kernels/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/kernels/utils.py b/unsloth/kernels/utils.py index 037c8c8a1f..d470c0f87e 100644 --- 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: