compressed-tensors 0.10.3a20250710__tar.gz → 0.10.3a20250715__tar.gz

{compressed_tensors-0.10.3a20250710/src/compressed_tensors.egg-info → compressed_tensors-0.10.3a20250715}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: compressed-tensors
-Version: 0.10.3a20250710
+Version: 0.10.3a20250715
 Summary: Library for utilization of compressed safetensors of neural network models
 Home-page: https://github.com/neuralmagic/compressed-tensors
 Author: Neuralmagic, Inc.

{compressed_tensors-0.10.3a20250710 → compressed_tensors-0.10.3a20250715}/src/compressed_tensors/compressors/model_compressors/model_compressor.py

@@ -747,12 +747,16 @@ class ModelCompressor:
 
 def map_module_to_scheme(model: Module) -> Dict[str, QuantizationScheme]:
     """
-    Returns a dictionary which maps quantized module names to their quantization schemes
+    Returns a dictionary which maps quantized module names to their quantization
+    schemes. Only includes modules with weight quantization
     """
     return {
         fix_fsdp_module_name(name): module.quantization_scheme
         for name, module in model.named_modules()
-        if is_module_quantized(module)
+        if (
+            hasattr(module, "quantization_scheme") and
+            module.quantization_scheme.weights is not None
+        )
     }
 
 

{compressed_tensors-0.10.3a20250710 → compressed_tensors-0.10.3a20250715}/src/compressed_tensors/transform/factory/base.py

@@ -99,7 +99,7 @@ class TransformFactory(RegistryMixin, ABC):
         :param args: defines how the transform will be applied to the target module
         """
         # create transform as submodule
-        transform_name = f"{self.name}_{args.location.value}"
+        transform_name = f"{self.name}_{args.location}"
         transform = self.create_transform(module, args)
         register_offload_module(module, transform_name, transform)
 

{compressed_tensors-0.10.3a20250710 → compressed_tensors-0.10.3a20250715}/src/compressed_tensors/transform/transform_args.py

@@ -46,7 +46,7 @@ class TransformLocation(str, Enum):
     Q_ATTN = "q_attn"
 
 
-class TransformArgs(BaseModel):
+class TransformArgs(BaseModel, use_enum_values=True):
     """
     Arguments which define *how* and where a transform should be applied to a model
 

{compressed_tensors-0.10.3a20250710 → compressed_tensors-0.10.3a20250715}/src/compressed_tensors/version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.10.3.a20250710'
+__version__ = version = '0.10.3.a20250715'
 __version_tuple__ = version_tuple = (0, 10, 3)

{compressed_tensors-0.10.3a20250710 → compressed_tensors-0.10.3a20250715/src/compressed_tensors.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: compressed-tensors
-Version: 0.10.3a20250710
+Version: 0.10.3a20250715
 Summary: Library for utilization of compressed safetensors of neural network models
 Home-page: https://github.com/neuralmagic/compressed-tensors
 Author: Neuralmagic, Inc.