rahul-tuli comment

neuralmagic · Nov 25, 2024 · 568a4ac · 568a4ac
1 parent 4665ca2
commit 568a4ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: