From 9eeede744f99c7c97054c7bb901fae1b3f5065cd Mon Sep 17 00:00:00 2001 From: Rahul Tuli Date: Tue, 17 Dec 2024 20:10:05 +0000 Subject: [PATCH] Update a docstring --- src/compressed_tensors/compressors/sparse_compressors/base.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/compressed_tensors/compressors/sparse_compressors/base.py b/src/compressed_tensors/compressors/sparse_compressors/base.py index c6b715de..0ecee029 100644 --- a/src/compressed_tensors/compressors/sparse_compressors/base.py +++ b/src/compressed_tensors/compressors/sparse_compressors/base.py @@ -129,7 +129,8 @@ def decompress( @staticmethod def should_compress(name: str, expanded_targets: Optional[Set[str]] = None) -> bool: """ - Check if a parameter should be compressed + Check if a parameter should be compressed. + Currently, this only returns True for weight parameters. :param name: name of the parameter :param expanded_targets: set of layer prefixes to compress