Ignore Dense sparsity config (#169)

neuralmagic · Sep 26, 2024 · 6ed94e6 · 6ed94e6
1 parent 44dda93
commit 6ed94e6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/compressed_tensors/compressors/model_compressor.py b/src/compressed_tensors/compressors/model_compressor.py
@@ -216,6 +216,11 @@ 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 = Compressor.load_from_registry(
                 sparsity_config.format, config=sparsity_config