fix conflict

neuralmagic · Mar 21, 2024 · b5fd814 · b5fd814
1 parent 5cdd162
commit b5fd814
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/sparseml/transformers/compression/compressors/sparse_bitmask.py b/src/sparseml/transformers/compression/compressors/sparse_bitmask.py
@@ -71,8 +71,6 @@ def compress(self, model_state: Dict[str, Tensor]) -> Dict[str, Tensor]:
                         "be replaced."
                     )
             compressed_dict |= bitmask_dict
-            bitmask_tensor = BitmaskTensor.from_dense(value)
-            compressed_dict |= bitmask_tensor.dict(name_prefix=name, device="cpu")
 
         return compressed_dict