From f53d7b990100bbc7b86dc4a497ff2037d260245f Mon Sep 17 00:00:00 2001 From: Sara Adkins Date: Mon, 9 Sep 2024 15:20:47 -0400 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com> --- src/transformers/quantizers/quantizer_compressed_tensors.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/quantizers/quantizer_compressed_tensors.py b/src/transformers/quantizers/quantizer_compressed_tensors.py index 6d18bcf220c757..5531838e568afb 100644 --- a/src/transformers/quantizers/quantizer_compressed_tensors.py +++ b/src/transformers/quantizers/quantizer_compressed_tensors.py @@ -29,7 +29,7 @@ class CompressedTensorsHfQuantizer(HfQuantizer): quantized state with compressed_tensors """ - requires_calibration = False + requires_calibration = True required_packages = ["compressed_tensors"] def __init__(self, quantization_config: QuantizationConfigMixin, **kwargs):