riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-18 20:18:24 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

[Ready to Merge][HFQuantizer] Squelch pydantic warnings (#37726)

Browse Source 
replace dict with model_dump

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
This commit is contained in:
Kyle Sayers 2025-05-05 14:38:49 -04:00 committed by GitHub
parent 46c0e1ff80
commit fe29b8c487
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/utils/quantization_config.py
View File

@ -1403,12 +1403,12 @@ class CompressedTensorsConfig(QuantizationConfigMixin):
""" """
quantization_config = {} quantization_config = {}
if self.quantization_config is not None: if self.quantization_config is not None:
quantization_config = self.quantization_config.dict() quantization_config = self.quantization_config.model_dump()
else: else:
quantization_config["quant_method"] = QuantizationMethod.COMPRESSED_TENSORS quantization_config["quant_method"] = QuantizationMethod.COMPRESSED_TENSORS
if self.sparsity_config is not None: if self.sparsity_config is not None:
quantization_config["sparsity_config"] = self.sparsity_config.dict() quantization_config["sparsity_config"] = self.sparsity_config.model_dump()
else: else:
quantization_config["sparsity_config"] = {} quantization_config["sparsity_config"] = {}