-
Notifications
You must be signed in to change notification settings - Fork 149
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.
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
[Quantization] Add alias mapping for target types (fixes SiLU name change issue) #2214
Conversation
@dbogunowicz to revert test patches in #2204 to test this PR |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This is a great solution, thanks
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@bfineran LGTM; testing confirms that. Let's hodl off with landing this patch as proposed during our call and wait for the parent PR to land first.
The base branch was changed.
transformers recently updated usage of
SiLU
to theSiLUActivation
class. When upgrading transformers this will cause backwards compatibility issues with existing recipes.This PR proposes a fix by adding class alias support in quantization modifier name matching. Any reference to
SiLU
in a recipe will now also check forSiLUActivation
test_plan:
@dbogunowicz to revert test patches in #2204 to test this PR