compressed-tensors 0.10.3a20250710__py3-none-any.whl → 0.10.3a20250711__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.
Files changed (7) hide show
  1. compressed_tensors/compressors/model_compressors/model_compressor.py +6 -2
  2. compressed_tensors/version.py +1 -1
  3. {compressed_tensors-0.10.3a20250710.dist-info → compressed_tensors-0.10.3a20250711.dist-info}/METADATA +1 -1
  4. {compressed_tensors-0.10.3a20250710.dist-info → compressed_tensors-0.10.3a20250711.dist-info}/RECORD +7 -7
  5. {compressed_tensors-0.10.3a20250710.dist-info → compressed_tensors-0.10.3a20250711.dist-info}/WHEEL +0 -0
  6. {compressed_tensors-0.10.3a20250710.dist-info → compressed_tensors-0.10.3a20250711.dist-info}/licenses/LICENSE +0 -0
  7. {compressed_tensors-0.10.3a20250710.dist-info → compressed_tensors-0.10.3a20250711.dist-info}/top_level.txt +0 -0
compressed_tensors/compressors/model_compressors/model_compressor.py CHANGED
@@ -747,12 +747,16 @@ class ModelCompressor:
747
747
 
748
748
  def map_module_to_scheme(model: Module) -> Dict[str, QuantizationScheme]:
749
749
  """
750
- Returns a dictionary which maps quantized module names to their quantization schemes
750
+ Returns a dictionary which maps quantized module names to their quantization
751
+ schemes. Only includes modules with weight quantization
751
752
  """
752
753
  return {
753
754
  fix_fsdp_module_name(name): module.quantization_scheme
754
755
  for name, module in model.named_modules()
755
- if is_module_quantized(module)
756
+ if (
757
+ hasattr(module, "quantization_scheme") and
758
+ module.quantization_scheme.weights is not None
759
+ )
756
760
  }
757
761
 
758
762
 
compressed_tensors/version.py CHANGED
@@ -17,5 +17,5 @@ __version__: str
17
17
  __version_tuple__: VERSION_TUPLE
18
18
  version_tuple: VERSION_TUPLE
19
19
 
20
- __version__ = version = '0.10.3.a20250710'
20
+ __version__ = version = '0.10.3.a20250711'
21
21
  __version_tuple__ = version_tuple = (0, 10, 3)
{compressed_tensors-0.10.3a20250710.dist-info → compressed_tensors-0.10.3a20250711.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: compressed-tensors
3
- Version: 0.10.3a20250710
3
+ Version: 0.10.3a20250711
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.
{compressed_tensors-0.10.3a20250710.dist-info → compressed_tensors-0.10.3a20250711.dist-info}/RECORD RENAMED
@@ -1,11 +1,11 @@
1
1
  compressed_tensors/__init__.py,sha256=UtKmifNeBCSE2TZSAfduVNNzHY-3V7bLjZ7n7RuXLOE,812
2
2
  compressed_tensors/base.py,sha256=73HYH7HY7O2roC89yG_piPFnZwrBfn_i7HmKl90SKc0,875
3
- compressed_tensors/version.py,sha256=gqWN6LSE0C4r0hiPn2KpmE4YPwpZnktZcE8p9BTZksQ,523
3
+ compressed_tensors/version.py,sha256=S11uf7PlZb4aZTrdFIT3CUNVegXSY80RCGN3qLFo2UQ,523
4
4
  compressed_tensors/compressors/__init__.py,sha256=smSygTSfcfuujRrAXDc6uZm4L_ccV1tWZewqVnOb4lM,825
5
5
  compressed_tensors/compressors/base.py,sha256=nvWsv4xEw1Tkxkxth6TmHplDYXfBeP22xWxOsZERyDY,7204
6
6
  compressed_tensors/compressors/helpers.py,sha256=OK6qxX9j3bHwF9JfIYSGMgBJe2PWjlTA3byXKCJaTIQ,5431
7
7
  compressed_tensors/compressors/model_compressors/__init__.py,sha256=5RGGPFu4YqEt_aOdFSQYFYFDjcZFJN0CsMqRtDZz3Js,666
8
- compressed_tensors/compressors/model_compressors/model_compressor.py,sha256=Zl33ceJu_KH_Vx2B5EK-bvE7W5OklOL0hb-QqdOVSsE,32906
8
+ compressed_tensors/compressors/model_compressors/model_compressor.py,sha256=MIMgjLK0VJu4bF_CwyvSftFrC5tvIIMfRt6XB0yLjOE,33056
9
9
  compressed_tensors/compressors/quantized_compressors/__init__.py,sha256=KvaFBL_Q84LxRGJOV035M8OBoCkAx8kOkfphswgkKWk,745
10
10
  compressed_tensors/compressors/quantized_compressors/base.py,sha256=YGUMzbxekj_36ChgQnVZN6T8uDjXtGG1zfMIBGBLWco,10354
11
11
  compressed_tensors/compressors/quantized_compressors/naive_quantized.py,sha256=0ANDcuD8aXPqTYNPY6GnX9iS6eXJw6P0TzNV_rYS2l8,5369
@@ -61,8 +61,8 @@ compressed_tensors/utils/permutations_24.py,sha256=kx6fsfDHebx94zsSzhXGyCyuC9sVy
61
61
  compressed_tensors/utils/permute.py,sha256=V6tJLKo3Syccj-viv4F7ZKZgJeCB-hl-dK8RKI_kBwI,2355
62
62
  compressed_tensors/utils/safetensors_load.py,sha256=DMfZBuUbA6qp_BG_zIWT3ckiEE33K9ob34s-OgzReO4,12057
63
63
  compressed_tensors/utils/semi_structured_conversions.py,sha256=XKNffPum54kPASgqKzgKvyeqWPAkair2XEQXjkp7ho8,13489
64
- compressed_tensors-0.10.3a20250710.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
65
- compressed_tensors-0.10.3a20250710.dist-info/METADATA,sha256=TodSeLplEMKQAyDzDHZDT7DPvFFEK5-qUokJdpw2yCg,7031
66
- compressed_tensors-0.10.3a20250710.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
67
- compressed_tensors-0.10.3a20250710.dist-info/top_level.txt,sha256=w2i-GyPs2s1UwVxvutSvN_lM22SXC2hQFBmoMcPnV7Y,19
68
- compressed_tensors-0.10.3a20250710.dist-info/RECORD,,
64
+ compressed_tensors-0.10.3a20250711.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
65
+ compressed_tensors-0.10.3a20250711.dist-info/METADATA,sha256=hKMAAe8C0-WWW8kZ40-9a3tEJy0gzknT3u9xQmww6w8,7031
66
+ compressed_tensors-0.10.3a20250711.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
67
+ compressed_tensors-0.10.3a20250711.dist-info/top_level.txt,sha256=w2i-GyPs2s1UwVxvutSvN_lM22SXC2hQFBmoMcPnV7Y,19
68
+ compressed_tensors-0.10.3a20250711.dist-info/RECORD,,