Skip to content

Commit 1ae2476

Browse files
Merge branch 'int8' of https://github.com/TimDettmers/bitsandbytes into int8
2 parents 0aefeb0 + bfb42d1 commit 1ae2476

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

bitsandbytes/functional.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1439,11 +1439,9 @@ def dequantize_4bit(
14391439
else:
14401440
raise ValueError(f"Blockwise quantization only supports 16/32-bit floats, but got {A.dtype}")
14411441

1442-
is_transposed = A.shape[0] == 1
1443-
if is_transposed:
1442+
if A.shape[0] == 1: # is transposed, transpose back
14441443
return out.t()
1445-
else:
1446-
return out
1444+
return out
14471445

14481446

14491447
@deprecated("This function is deprecated and will be removed in a future release.", category=FutureWarning)

0 commit comments

Comments
 (0)