Skip to content

Commit 6fffbd7

Browse files
authored
Fix: Disable Sparse Decompression for Dense Compressors (#237)
* Turn off sparse decompression when sparse compressor is dense * Update: Condiition to use enum instead of raw string * Remove unnecesarry style change
1 parent f0c369a commit 6fffbd7

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/compressed_tensors/compressors/model_compressors/model_compressor.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -310,7 +310,10 @@ def decompress(self, model_path: str, model: Module):
310310
model_path = get_safetensors_folder(model_path)
311311
sparse_decompressed = False
312312

313-
if self.sparsity_compressor is not None:
313+
if (
314+
self.sparsity_compressor is not None
315+
and self.sparsity_config.format != CompressionFormat.dense.value
316+
):
314317
# Sparse decompression is applied on the model_path
315318
dense_gen = self.sparsity_compressor.decompress(model_path)
316319
self._replace_weights(dense_gen, model)

0 commit comments

Comments
 (0)