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: