From bfb42d1b72f5aa32ea2f1387020542774474dca7 Mon Sep 17 00:00:00 2001 From: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com> Date: Mon, 4 Nov 2024 12:39:38 -0500 Subject: [PATCH] Update bitsandbytes/functional.py Co-authored-by: Aarni Koskela --- bitsandbytes/functional.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/bitsandbytes/functional.py b/bitsandbytes/functional.py index 536786f41..2a0f38510 100644 --- a/bitsandbytes/functional.py +++ b/bitsandbytes/functional.py @@ -1439,11 +1439,9 @@ def dequantize_4bit( else: raise ValueError(f"Blockwise quantization only supports 16/32-bit floats, but got {A.dtype}") - is_transposed = A.shape[0] == 1 - if is_transposed: + if A.shape[0] == 1: # is transposed, transpose back return out.t() - else: - return out + return out @deprecated("This function is deprecated and will be removed in a future release.", category=FutureWarning)