diff --git a/src/brevitas_examples/llm/main.py b/src/brevitas_examples/llm/main.py index 63898d6ef..b9c2c1c4d 100644 --- a/src/brevitas_examples/llm/main.py +++ b/src/brevitas_examples/llm/main.py @@ -399,7 +399,7 @@ def main(args): with torch.no_grad(): model(**calibration_loader[0]) - # We restore the original behaviour of the post-forward. + # We restore the original behaviour of the post-forward. for k, v in dict_hooks.items(): k._hf_hook.post_forward = v