Update src/llmcompressor/modifiers/quantization/calibration.py

shanjiaz · kylesayrs · web-flow · commit c0844b200063 · 2025-07-23T11:42:41.000-04:00
Co-authored-by: Kyle Sayers &lt;kylesayrs@gmail.com&gt;
diff --git a/src/llmcompressor/modifiers/quantization/calibration.py b/src/llmcompressor/modifiers/quantization/calibration.py
@@ -141,7 +141,7 @@ def call_observer(
                 ):
                     if hasattr(module, name):
                         delete_offload_parameter(module, name)
-                    module.register_offload_parameter(
+                    register_offload_parameter(module
                         name, torch.nn.Parameter(value.clone(), requires_grad=False)
                     )
                 else:

Original file line number	Diff line number	Diff line change
`@@ -141,7 +141,7 @@ def call_observer(`
`141`	`141`	`):`
`142`	`142`	`if hasattr(module, name):`
`143`	`143`	`delete_offload_parameter(module, name)`
`144`		`- module.register_offload_parameter(`
	`144`	`+ register_offload_parameter(module`
`145`	`145`	`name, torch.nn.Parameter(value.clone(), requires_grad=False)`
`146`	`146`	`)`
`147`	`147`	`else:`