From c2455b7c5040adc0e7fc6a197144971b2597d607 Mon Sep 17 00:00:00 2001 From: Rahul Tuli Date: Fri, 4 Oct 2024 16:07:11 -0400 Subject: [PATCH] Revert "Ignore Dense sparsity config (#169)" (#181) --- .../compressors/model_compressors/model_compressor.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/src/compressed_tensors/compressors/model_compressors/model_compressor.py b/src/compressed_tensors/compressors/model_compressors/model_compressor.py index ac15fdaa..6473554d 100644 --- a/src/compressed_tensors/compressors/model_compressors/model_compressor.py +++ b/src/compressed_tensors/compressors/model_compressors/model_compressor.py @@ -242,10 +242,6 @@ def __init__( self.sparsity_compressor = None self.quantization_compressor = None - if sparsity_config and sparsity_config.format == CompressionFormat.dense.value: - # ignore dense sparsity config - self.sparsity_config = None - if sparsity_config is not None: self.sparsity_compressor = BaseCompressor.load_from_registry( sparsity_config.format, config=sparsity_config