We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 735660d commit 6017f05Copy full SHA for 6017f05
src/compressed_tensors/compressors/quantized_compressors/base.py
@@ -76,8 +76,7 @@ def compress(
76
"""
77
Compresses a dense state dict
78
79
- :param model_state: state dict of uncompressed model, consumed during
80
- compression
+ :param model_state: state dict of uncompressed model, consumed by compression
81
:param names_to_scheme: quantization args for each quantized weight, needed for
82
quantize function to calculate bit depth
83
:return: compressed state dict
0 commit comments