We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1534861 commit f6e0dfbCopy full SHA for f6e0dfb
src/compressed_tensors/quantization/utils/helpers.py
@@ -205,7 +205,7 @@ def calculate_range(
205
:return: tuple endpoints for the given quantization range
206
"""
207
if quantization_args.type == QuantizationType.INT:
208
- bit_range = 2.0 ** quantization_args.num_bits
+ bit_range = 2.0**quantization_args.num_bits
209
q_max = torch.tensor(bit_range / 2 - 1, device=device)
210
q_min = torch.tensor(-bit_range / 2, device=device)
211
elif quantization_args.type == QuantizationType.FLOAT:
0 commit comments