compressed-tensors-nightly 0.9.0.20250120__py3-none-any.whl → 0.9.0.20250122__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- compressed_tensors/compressors/sparse_compressors/sparse_24_bitmask.py +4 -2
- compressed_tensors/utils/helpers.py +1 -1
- {compressed_tensors_nightly-0.9.0.20250120.dist-info → compressed_tensors_nightly-0.9.0.20250122.dist-info}/METADATA +1 -1
- {compressed_tensors_nightly-0.9.0.20250120.dist-info → compressed_tensors_nightly-0.9.0.20250122.dist-info}/RECORD +7 -7
- {compressed_tensors_nightly-0.9.0.20250120.dist-info → compressed_tensors_nightly-0.9.0.20250122.dist-info}/LICENSE +0 -0
- {compressed_tensors_nightly-0.9.0.20250120.dist-info → compressed_tensors_nightly-0.9.0.20250122.dist-info}/WHEEL +0 -0
- {compressed_tensors_nightly-0.9.0.20250120.dist-info → compressed_tensors_nightly-0.9.0.20250122.dist-info}/top_level.txt +0 -0
|
@@ -103,8 +103,10 @@ class Sparse24BitMaskTensor:
|
|
|
103
103
|
:param bitmask: 2d bitmask of non-zero values
|
|
104
104
|
:return: instantiated Sparse24BitMaskTensor
|
|
105
105
|
"""
|
|
106
|
-
if isinstance(shape,
|
|
107
|
-
shape =
|
|
106
|
+
if isinstance(shape, list):
|
|
107
|
+
shape = torch.tensor(shape)
|
|
108
|
+
if isinstance(shape, torch.Tensor):
|
|
109
|
+
shape = shape.flatten().tolist()
|
|
108
110
|
return Sparse24BitMaskTensor(
|
|
109
111
|
shape=shape, compressed=compressed, bitmask=bitmask
|
|
110
112
|
)
|
|
@@ -301,7 +301,7 @@ def pack_bitmasks(bytemasks: torch.Tensor) -> torch.Tensor:
|
|
|
301
301
|
|
|
302
302
|
|
|
303
303
|
def unpack_bitmasks(
|
|
304
|
-
packed_bitmasks: torch.Tensor, original_shape:
|
|
304
|
+
packed_bitmasks: torch.Tensor, original_shape: List[int]
|
|
305
305
|
) -> torch.Tensor:
|
|
306
306
|
"""
|
|
307
307
|
Converts a bitmask tensor back to a bytemask tensor for use during decompression
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
2
|
Name: compressed-tensors-nightly
|
|
3
|
-
Version: 0.9.0.
|
|
3
|
+
Version: 0.9.0.20250122
|
|
4
4
|
Summary: Library for utilization of compressed safetensors of neural network models
|
|
5
5
|
Home-page: https://github.com/neuralmagic/compressed-tensors
|
|
6
6
|
Author: Neuralmagic, Inc.
|
|
@@ -13,7 +13,7 @@ compressed_tensors/compressors/quantized_compressors/pack_quantized.py,sha256=OO
|
|
|
13
13
|
compressed_tensors/compressors/sparse_compressors/__init__.py,sha256=Atuz-OdEgn8OCUhx7Ovd6gXdyImAI186uCR-uR0t_Nk,737
|
|
14
14
|
compressed_tensors/compressors/sparse_compressors/base.py,sha256=9e841MQWr0j8m33ejDw_jP5_BIpQ5099x9_pvuZ-Nr0,5944
|
|
15
15
|
compressed_tensors/compressors/sparse_compressors/dense.py,sha256=lSKNWRx6H7aUqaJj1j4qbXk8Gkm1UohbnvW1Rvq6Ra4,1284
|
|
16
|
-
compressed_tensors/compressors/sparse_compressors/sparse_24_bitmask.py,sha256=
|
|
16
|
+
compressed_tensors/compressors/sparse_compressors/sparse_24_bitmask.py,sha256=_g139pe4iAFn5jvGIEk4v-qMoyx9ID6E88vriPSNYV4,8604
|
|
17
17
|
compressed_tensors/compressors/sparse_compressors/sparse_bitmask.py,sha256=7zSr9bqkpuH1ivQpxtYBNxXIoElal7Jo1nSKpZN_IFk,5633
|
|
18
18
|
compressed_tensors/compressors/sparse_quantized_compressors/__init__.py,sha256=4f_cwcKXB1nVVMoiKgTFAc8jAPjPLElo-Df_EDm1_xw,675
|
|
19
19
|
compressed_tensors/compressors/sparse_quantized_compressors/marlin_24.py,sha256=BMIQWTLlnUvxy14iEJegtiP75WHJeOVojey9mKOK1hE,9427
|
|
@@ -39,14 +39,14 @@ compressed_tensors/quantization/utils/helpers.py,sha256=DBP-sGRpGAY01K0LFE7qqonN
|
|
|
39
39
|
compressed_tensors/registry/__init__.py,sha256=FwLSNYqfIrb5JD_6OK_MT4_svvKTN_nEhpgQlQvGbjI,658
|
|
40
40
|
compressed_tensors/registry/registry.py,sha256=vRcjVB1ITfSbfYUaGndBBmqhip_5vsS62weorVg0iXo,11896
|
|
41
41
|
compressed_tensors/utils/__init__.py,sha256=gS4gSU2pwcAbsKj-6YMaqhm25udFy6ISYaWBf-myRSM,808
|
|
42
|
-
compressed_tensors/utils/helpers.py,sha256=
|
|
42
|
+
compressed_tensors/utils/helpers.py,sha256=xQHZXwIAAybC8mMTiAtWSOeggZMT1JOC6_wcDvlo2yk,10320
|
|
43
43
|
compressed_tensors/utils/offload.py,sha256=cMmzd9IdlNbs29CReHj1PPSLUM6OWaT5YumlLT5eP3w,13845
|
|
44
44
|
compressed_tensors/utils/permutations_24.py,sha256=kx6fsfDHebx94zsSzhXGyCyuC9sVyah6BUUir_StT28,2530
|
|
45
45
|
compressed_tensors/utils/permute.py,sha256=V6tJLKo3Syccj-viv4F7ZKZgJeCB-hl-dK8RKI_kBwI,2355
|
|
46
46
|
compressed_tensors/utils/safetensors_load.py,sha256=fBuoHVPoBt1mkvqFJ60zQIASX_4nhl0-6QfFS27NY8I,11430
|
|
47
47
|
compressed_tensors/utils/semi_structured_conversions.py,sha256=XKNffPum54kPASgqKzgKvyeqWPAkair2XEQXjkp7ho8,13489
|
|
48
|
-
compressed_tensors_nightly-0.9.0.
|
|
49
|
-
compressed_tensors_nightly-0.9.0.
|
|
50
|
-
compressed_tensors_nightly-0.9.0.
|
|
51
|
-
compressed_tensors_nightly-0.9.0.
|
|
52
|
-
compressed_tensors_nightly-0.9.0.
|
|
48
|
+
compressed_tensors_nightly-0.9.0.20250122.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
|
|
49
|
+
compressed_tensors_nightly-0.9.0.20250122.dist-info/METADATA,sha256=OS8Ss5w7aa8djfWC7QwCVBlsnI8nsudxhVmpraj00iA,6799
|
|
50
|
+
compressed_tensors_nightly-0.9.0.20250122.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
|
|
51
|
+
compressed_tensors_nightly-0.9.0.20250122.dist-info/top_level.txt,sha256=w2i-GyPs2s1UwVxvutSvN_lM22SXC2hQFBmoMcPnV7Y,19
|
|
52
|
+
compressed_tensors_nightly-0.9.0.20250122.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|