Skip to content

Commit 0e9544d

Browse files
committed
update docstring
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent cbb258b commit 0e9544d

File tree

1 file changed

+2
-0
lines changed
  • src/compressed_tensors/compressors/sparse_compressors

1 file changed

+2
-0
lines changed

src/compressed_tensors/compressors/sparse_compressors/base.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,8 @@ def decompress_from_state_dict(
160160
state_dict: Dict[str, Tensor],
161161
) -> Generator[Tuple[str, Dict[str, Tensor]], None, None]:
162162
"""
163+
Decompress the state dict of a module (or model)
164+
163165
Unlike `self.decompress`, this function does not need to explicitly skip params
164166
via params_to_skip_load because it is more convenient for its only caller
165167
(ModelCompressor.decompress_model) to retrieve all unused param keys

0 commit comments

Comments
 (0)