Skip to content

Commit

Permalink
precommit
Browse files Browse the repository at this point in the history
  • Loading branch information
Giuseppe5 committed Dec 17, 2024
1 parent 7b68194 commit 5190deb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/brevitas_examples/llm/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit 5190deb

Please sign in to comment.