diff --git a/src/compressed_tensors/compressors/model_compressors/model_compressor.py b/src/compressed_tensors/compressors/model_compressors/model_compressor.py index 4e24b2fb..78ab6422 100644 --- 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)