Update src/compressed_tensors/compressors/sparse_compressors/sparse_24_bitmask.py

shanjiaz · brian-dellabetta · web-flow · commit 4a5f064f64b6 · 2025-07-07T10:41:02.000-04:00
Co-authored-by: Brian Dellabetta &lt;brian-dellabetta@users.noreply.github.com&gt;
diff --git a/src/compressed_tensors/compressors/sparse_compressors/sparse_24_bitmask.py b/src/compressed_tensors/compressors/sparse_compressors/sparse_24_bitmask.py
@@ -58,7 +58,7 @@ def compress_weight(self, name, value):
         )
         return bitmask_tensor.dict(
             name_prefix=name,
-            device="meta" if value.device.type == "meta" else "cpu",
+            device="meta" if value.is_meta else "cpu",
         )
 
     def decompress_weight(self, weight_data):

Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@ def compress_weight(self, name, value):`
`58`	`58`	`)`
`59`	`59`	`return bitmask_tensor.dict(`
`60`	`60`	`name_prefix=name,`
`61`		`- device="meta" if value.device.type == "meta" else "cpu",`
	`61`	`+ device="meta" if value.is_meta else "cpu",`
`62`	`62`	`)`
`63`	`63`
`64`	`64`	`def decompress_weight(self, weight_data):`