Skip to content

Commit 6017f05

Browse files
committed
update docstring
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent 735660d commit 6017f05

File tree

1 file changed

+1
-2
lines changed
  • src/compressed_tensors/compressors/quantized_compressors

1 file changed

+1
-2
lines changed

src/compressed_tensors/compressors/quantized_compressors/base.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,8 +76,7 @@ def compress(
7676
"""
7777
Compresses a dense state dict
7878
79-
:param model_state: state dict of uncompressed model, consumed during
80-
compression
79+
:param model_state: state dict of uncompressed model, consumed by compression
8180
:param names_to_scheme: quantization args for each quantized weight, needed for
8281
quantize function to calculate bit depth
8382
:return: compressed state dict

0 commit comments

Comments
 (0)