Skip to content
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

Merged
merged 15 commits into from
Apr 3, 2024

Conversation

bfineran
Copy link
Contributor

@bfineran bfineran commented Apr 3, 2024

transformers recently updated usage of SiLU to the SiLUActivation 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 for SiLUActivation

test_plan:
@dbogunowicz to revert test patches in #2204 to test this PR

@bfineran bfineran requested a review from dbogunowicz April 3, 2024 14:16
@bfineran bfineran self-assigned this Apr 3, 2024
@bfineran
Copy link
Contributor Author

bfineran commented Apr 3, 2024

@dbogunowicz to revert test patches in #2204 to test this PR

mgoin
mgoin previously approved these changes Apr 3, 2024
Copy link
Member

@mgoin mgoin left a 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

dbogunowicz
dbogunowicz previously approved these changes Apr 3, 2024
Copy link
Contributor

@dbogunowicz dbogunowicz left a 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.

Base automatically changed from feature/damian/update_transformers to main April 3, 2024 15:54
@dbogunowicz dbogunowicz dismissed stale reviews from mgoin and themself April 3, 2024 15:54

The base branch was changed.

@dbogunowicz dbogunowicz requested a review from mgoin April 3, 2024 16:14
@bfineran bfineran merged commit 3b813b6 into main Apr 3, 2024
13 of 15 checks passed
@bfineran bfineran deleted the quantization-name-match-patch branch April 3, 2024 16:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants