fix typo (#241)

Author: rahul-tuli

src/compressed_tensors/compressors/sparse_compressors/sparse_24_bitmask.py

@@ -103,8 +103,10 @@ def from_compressed_data(
         :param bitmask: 2d bitmask of non-zero values
         :return: instantiated Sparse24BitMaskTensor
         """
-        if isinstance(shape, Tensor):
-            shape = shape.tolist()
+        if isinstance(shape, list):
+            shape = torch.tensor(shape)
+        if isinstance(shape, torch.Tensor):
+            shape = shape.flatten().tolist()
         return Sparse24BitMaskTensor(
             shape=shape, compressed=compressed, bitmask=bitmask
         )

src/compressed_tensors/utils/helpers.py

@@ -301,7 +301,7 @@ def pack_bitmasks(bytemasks: torch.Tensor) -> torch.Tensor:
 
 
 def unpack_bitmasks(
-    packed_bitmasks: torch.Tensor, original_shape: torch.Size
+    packed_bitmasks: torch.Tensor, original_shape: List[int]
 ) -> torch.Tensor:
     """
     Converts a bitmask tensor back to a bytemask tensor for use during decompression