New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update signature check #904

Open

Giuseppe5 opened this issue Mar 12, 2024 · 0 comments

Collaborator

Giuseppe5 commented Mar 12, 2024

brevitas/src/brevitas/nn/equalized_layer.py

Line 19 in 2004568

bound_arguments = signature(self.layer.forward).bind(*args, **kwargs)

Has to be done similarly to:

brevitas/src/brevitas/graph/equalize.py

Line 639 in 2004568

forward_to_check = module._old_forward if hasattr(

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment