compressed-tensors 0.9.0__tar.gz → 0.9.1__tar.gz

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 (58) hide show
  1. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/PKG-INFO +1 -1
  2. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/sparse_compressors/sparse_24_bitmask.py +4 -2
  3. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/utils/helpers.py +1 -1
  4. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/version.py +1 -1
  5. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors.egg-info/PKG-INFO +1 -1
  6. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/LICENSE +0 -0
  7. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/README.md +0 -0
  8. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/pyproject.toml +0 -0
  9. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/setup.cfg +0 -0
  10. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/setup.py +0 -0
  11. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/__init__.py +0 -0
  12. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/base.py +0 -0
  13. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/__init__.py +0 -0
  14. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/base.py +0 -0
  15. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/helpers.py +0 -0
  16. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/model_compressors/__init__.py +0 -0
  17. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/model_compressors/model_compressor.py +0 -0
  18. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/quantized_compressors/__init__.py +0 -0
  19. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/quantized_compressors/base.py +0 -0
  20. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/quantized_compressors/naive_quantized.py +0 -0
  21. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/quantized_compressors/pack_quantized.py +0 -0
  22. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/sparse_compressors/__init__.py +0 -0
  23. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/sparse_compressors/base.py +0 -0
  24. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/sparse_compressors/dense.py +0 -0
  25. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/sparse_compressors/sparse_bitmask.py +0 -0
  26. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/sparse_quantized_compressors/__init__.py +0 -0
  27. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/sparse_quantized_compressors/marlin_24.py +0 -0
  28. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/config/__init__.py +0 -0
  29. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/config/base.py +0 -0
  30. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/config/dense.py +0 -0
  31. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/config/sparse_24_bitmask.py +0 -0
  32. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/config/sparse_bitmask.py +0 -0
  33. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/linear/__init__.py +0 -0
  34. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/linear/compressed_linear.py +0 -0
  35. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/quantization/__init__.py +0 -0
  36. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/quantization/lifecycle/__init__.py +0 -0
  37. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/quantization/lifecycle/apply.py +0 -0
  38. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/quantization/lifecycle/compressed.py +0 -0
  39. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/quantization/lifecycle/forward.py +0 -0
  40. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/quantization/lifecycle/helpers.py +0 -0
  41. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/quantization/lifecycle/initialize.py +0 -0
  42. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/quantization/quant_args.py +0 -0
  43. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/quantization/quant_config.py +0 -0
  44. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/quantization/quant_scheme.py +0 -0
  45. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/quantization/utils/__init__.py +0 -0
  46. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/quantization/utils/helpers.py +0 -0
  47. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/registry/__init__.py +0 -0
  48. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/registry/registry.py +0 -0
  49. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/utils/__init__.py +0 -0
  50. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/utils/offload.py +0 -0
  51. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/utils/permutations_24.py +0 -0
  52. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/utils/permute.py +0 -0
  53. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/utils/safetensors_load.py +0 -0
  54. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/utils/semi_structured_conversions.py +0 -0
  55. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors.egg-info/SOURCES.txt +0 -0
  56. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors.egg-info/dependency_links.txt +0 -0
  57. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors.egg-info/requires.txt +0 -0
  58. {compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors.egg-info/top_level.txt +0 -0
{compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: compressed-tensors
3
- Version: 0.9.0
3
+ Version: 0.9.1
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.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/compressors/sparse_compressors/sparse_24_bitmask.py RENAMED
@@ -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, Tensor):
107
- shape = shape.tolist()
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
  )
{compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/utils/helpers.py RENAMED
@@ -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: torch.Size
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
{compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors/version.py RENAMED
@@ -17,7 +17,7 @@ Functionality for storing and setting the version info for SparseML
17
17
  """
18
18
 
19
19
 
20
- version_base = "0.9.0"
20
+ version_base = "0.9.1"
21
21
  is_release = True # change to True to set the generated version as a release version
22
22
 
23
23
 
{compressed-tensors-0.9.0 → compressed-tensors-0.9.1}/src/compressed_tensors.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: compressed-tensors
3
- Version: 0.9.0
3
+ Version: 0.9.1
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.