Skip to content

Commit

Permalink
Fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
dsikka committed Oct 9, 2024
1 parent c85208c commit 3ef9db9
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ def set_module_for_calibration(module: Module, quantize_weights_upfront: bool =
if not hasattr(module, "weight_observer"):
initialize_observers(
module=module,
base_name="weight_observer",
base_name="weight",
quantization_args=module.quantization_scheme.weights,
)
observer = module.weight_observer
Expand Down

0 comments on commit 3ef9db9

Please sign in to comment.