compressed-tensors-nightly 0.8.1.20241216__tar.gz → 0.8.1.20241217__tar.gz

{compressed-tensors-nightly-0.8.1.20241216/src/compressed_tensors_nightly.egg-info → compressed-tensors-nightly-0.8.1.20241217}/PKG-INFO

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

{compressed-tensors-nightly-0.8.1.20241216 → compressed-tensors-nightly-0.8.1.20241217}/src/compressed_tensors/compressors/model_compressors/model_compressor.py

@@ -137,7 +137,7 @@ class ModelCompressor:
                 format, **sparsity_config
             )
         if quantization_config is not None:
-            quantization_config = QuantizationConfig.parse_obj(quantization_config)
+            quantization_config = QuantizationConfig.model_validate(quantization_config)
 
         return cls(
             sparsity_config=sparsity_config, quantization_config=quantization_config
@@ -193,7 +193,7 @@ class ModelCompressor:
 
         if is_compressed_tensors_config(compression_config):
             s_config = compression_config.sparsity_config
-            return s_config.dict() if s_config is not None else None
+            return s_config.model_dump() if s_config is not None else None
 
         return compression_config.get(SPARSITY_CONFIG_NAME, None)
 
@@ -214,7 +214,7 @@ class ModelCompressor:
 
         if is_compressed_tensors_config(compression_config):
             q_config = compression_config.quantization_config
-            return q_config.dict() if q_config is not None else None
+            return q_config.model_dump() if q_config is not None else None
 
         quantization_config = deepcopy(compression_config)
         quantization_config.pop(SPARSITY_CONFIG_NAME, None)

{compressed-tensors-nightly-0.8.1.20241216 → compressed-tensors-nightly-0.8.1.20241217}/src/compressed_tensors/quantization/quant_config.py

@@ -160,7 +160,7 @@ class QuantizationConfig(BaseModel):
 
     def to_dict(self):
         # for compatibility with HFQuantizer
-        return self.dict()
+        return self.model_dump()
 
     @staticmethod
     def from_pretrained(

{compressed-tensors-nightly-0.8.1.20241216 → compressed-tensors-nightly-0.8.1.20241217/src/compressed_tensors_nightly.egg-info}/PKG-INFO

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