riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-04 05:10:06 +06:00
Code Issues Actions 20 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 = {}
if self.quantization_config is not None:
quantization_config = self.quantization_config.dict()
quantization_config = self.quantization_config.model_dump()
else:
quantization_config["quant_method"] = QuantizationMethod.COMPRESSED_TENSORS
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:
quantization_config["sparsity_config"] = {}