compressed-tensors 0.9.5a20250512__py3-none-any.whl → 0.9.5a20250513__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.
@@ -206,7 +206,7 @@ class QuantizationArgs(BaseModel, use_enum_values=True):
206
206
  return value
207
207
 
208
208
  @model_validator(mode="after")
209
- def validate_model_after(model: "QuantizationArgs") -> Dict[str, Any]:
209
+ def validate_model_after(model: "QuantizationArgs") -> "QuantizationArgs":
210
210
  # extract user-passed values from dictionary
211
211
  strategy = model.strategy
212
212
  group_size = model.group_size
@@ -48,7 +48,7 @@ class QuantizationScheme(BaseModel):
48
48
  output_activations: Optional[QuantizationArgs] = None
49
49
 
50
50
  @model_validator(mode="after")
51
- def validate_model_after(model: "QuantizationArgs") -> Dict[str, Any]:
51
+ def validate_model_after(model: "QuantizationScheme") -> "QuantizationScheme":
52
52
  inputs = model.input_activations
53
53
  outputs = model.output_activations
54
54
 
@@ -17,5 +17,5 @@ __version__: str
17
17
  __version_tuple__: VERSION_TUPLE
18
18
  version_tuple: VERSION_TUPLE
19
19
 
20
- __version__ = version = '0.9.5.a20250512'
20
+ __version__ = version = '0.9.5.a20250513'
21
21
  __version_tuple__ = version_tuple = (0, 9, 5)
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: compressed-tensors
3
- Version: 0.9.5a20250512
3
+ Version: 0.9.5a20250513
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.
@@ -1,6 +1,6 @@
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=Yd5MPtXGvm9XWCPM_O99KCK_9DwKmcl2OenuJ4MxlUI,521
3
+ compressed_tensors/version.py,sha256=anvjogcrhtUkFpir0cHcNUxTI9qeqiV_ELg-b1mW8Xc,521
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
@@ -26,9 +26,9 @@ compressed_tensors/config/sparse_bitmask.py,sha256=pZUboRNZTu6NajGOQEFExoPknak5y
26
26
  compressed_tensors/linear/__init__.py,sha256=fH6rjBYAxuwrTzBTlTjTgCYNyh6TCvCqajCz4Im4YrA,617
27
27
  compressed_tensors/linear/compressed_linear.py,sha256=_m6XpNcI53eeSHO8VdiuAM6UBTdpDhn5Ivd8iRMwEKc,3980
28
28
  compressed_tensors/quantization/__init__.py,sha256=83J5bPB7PavN2TfCoW7_vEDhfYpm4TDrqYO9vdSQ5bk,760
29
- compressed_tensors/quantization/quant_args.py,sha256=2m4WJWBnNjkU-3rVR_f2a6p_BZMvGfMrPyOui8JUwWk,10487
29
+ compressed_tensors/quantization/quant_args.py,sha256=CepGBAURFGxBzTyFXxHwsUs6wYEJ46_jPbEvJYMG0Tw,10491
30
30
  compressed_tensors/quantization/quant_config.py,sha256=MxSUcb5dOqMN6LFyD5K2h8X0TvEtcWIAoiUJqD2dHGE,10159
31
- compressed_tensors/quantization/quant_scheme.py,sha256=0FpN3R7bVn8rQ18Vp0NuDVpoilTZ7X8vk9zp_8AndwY,6578
31
+ compressed_tensors/quantization/quant_scheme.py,sha256=Fx7Ma4bDlFB6OWkHKhOB6_0AOVIOPRgNE_qTwmDLSbc,6586
32
32
  compressed_tensors/quantization/lifecycle/__init__.py,sha256=_uItzFWusyV74Zco_pHLOTdE9a83cL-R-ZdyQrBkIyw,772
33
33
  compressed_tensors/quantization/lifecycle/apply.py,sha256=DOoxH4jM8r0270GGGUFOpRrgwaisiJi7TV-Q6E8qM8E,18067
34
34
  compressed_tensors/quantization/lifecycle/compressed.py,sha256=Fj9n66IN0EWsOAkBHg3O0GlOQpxstqjCcs0ttzMXrJ0,2296
@@ -46,8 +46,8 @@ compressed_tensors/utils/permutations_24.py,sha256=kx6fsfDHebx94zsSzhXGyCyuC9sVy
46
46
  compressed_tensors/utils/permute.py,sha256=V6tJLKo3Syccj-viv4F7ZKZgJeCB-hl-dK8RKI_kBwI,2355
47
47
  compressed_tensors/utils/safetensors_load.py,sha256=kkkUDmS1H40MFy6FDP-DFGiAYbtqke6bKE7YrAtORtA,11499
48
48
  compressed_tensors/utils/semi_structured_conversions.py,sha256=XKNffPum54kPASgqKzgKvyeqWPAkair2XEQXjkp7ho8,13489
49
- compressed_tensors-0.9.5a20250512.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
50
- compressed_tensors-0.9.5a20250512.dist-info/METADATA,sha256=gArKa7gy0jdBGF5PbuNLVh_ZmnXq4CdEp1-7grxpjnw,7004
51
- compressed_tensors-0.9.5a20250512.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
52
- compressed_tensors-0.9.5a20250512.dist-info/top_level.txt,sha256=w2i-GyPs2s1UwVxvutSvN_lM22SXC2hQFBmoMcPnV7Y,19
53
- compressed_tensors-0.9.5a20250512.dist-info/RECORD,,
49
+ compressed_tensors-0.9.5a20250513.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
50
+ compressed_tensors-0.9.5a20250513.dist-info/METADATA,sha256=gMyJF3nQLJu7DCEbM0LDyD2GeuKwu6zxewO6qST7_s8,7004
51
+ compressed_tensors-0.9.5a20250513.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
52
+ compressed_tensors-0.9.5a20250513.dist-info/top_level.txt,sha256=w2i-GyPs2s1UwVxvutSvN_lM22SXC2hQFBmoMcPnV7Y,19
53
+ compressed_tensors-0.9.5a20250513.dist-info/RECORD,,