From 8d9de7af1551d226f300825f47451c3e309eb253 Mon Sep 17 00:00:00 2001 From: shanjiaz <43143795+shanjiaz@users.noreply.github.com> Date: Wed, 25 Jun 2025 14:37:25 -0400 Subject: [PATCH] Update src/transformers/quantizers/quantizer_compressed_tensors.py Co-authored-by: Kyle Sayers --- 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 e0fc1b9c535..b99aea4ba60 100644 --- a/src/transformers/quantizers/quantizer_compressed_tensors.py +++ b/src/transformers/quantizers/quantizer_compressed_tensors.py @@ -134,7 +134,7 @@ class CompressedTensorsHfQuantizer(HfQuantizer): quant_targets.update(group.targets) # Disable gradient computation for quantized int modules - for _, module in model.named_modules(): + for module in model.modules(): if type(module).__name__ in quant_targets: for param in module.parameters(): param.requires_grad = False