We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 0aefeb0 + bfb42d1 commit 1ae2476Copy full SHA for 1ae2476
bitsandbytes/functional.py
@@ -1439,11 +1439,9 @@ def dequantize_4bit(
1439
else:
1440
raise ValueError(f"Blockwise quantization only supports 16/32-bit floats, but got {A.dtype}")
1441
1442
- is_transposed = A.shape[0] == 1
1443
- if is_transposed:
+ if A.shape[0] == 1: # is transposed, transpose back
1444
return out.t()
1445
- else:
1446
- return out
+ return out
1447
1448
1449
@deprecated("This function is deprecated and will be removed in a future release.", category=FutureWarning)
0 commit comments