From 568a4ac9dc3e8f56a538c23ed887080110d37103 Mon Sep 17 00:00:00 2001 From: George Date: Mon, 25 Nov 2024 11:45:37 -0500 Subject: [PATCH] rahul-tuli comment --- .../compressors/quantized_compressors/pack_quantized.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/compressed_tensors/compressors/quantized_compressors/pack_quantized.py b/src/compressed_tensors/compressors/quantized_compressors/pack_quantized.py index 26026b87..3bbddc7c 100644 --- a/src/compressed_tensors/compressors/quantized_compressors/pack_quantized.py +++ b/src/compressed_tensors/compressors/quantized_compressors/pack_quantized.py @@ -149,7 +149,7 @@ def pack_to_int32(value: torch.Tensor, num_bits: int) -> torch.Tensor: 33409 -> 33409 :param value: tensor to pack - :param num_bits: number of bits used to store underlying data + :param num_bits: number of bits used to store underlying data, must be at least 1 :returns: packed int32 tensor """ if value.dtype is not torch.int8: