Update: Condiition to use enum instead of raw string

neuralmagic · Jan 10, 2025 · 91aa19d · 91aa19d
1 parent 8163759
commit 91aa19d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/compressed_tensors/compressors/model_compressors/model_compressor.py b/src/compressed_tensors/compressors/model_compressors/model_compressor.py
@@ -312,7 +312,7 @@ def decompress(self, model_path: str, model: Module):
 
         if (
             self.sparsity_compressor is not None
-            and self.sparsity_config.format != "dense"
+            and self.sparsity_config.format != CompressionFormat.dense.value
         ):
             # Sparse decompression is applied on the model_path
             dense_gen = self.sparsity_compressor.decompress(model_path)