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

Forkless transformers [4.34.1] #2199

Merged
merged 19 commits into from
Mar 29, 2024
Merged

Forkless transformers [4.34.1] #2199

merged 19 commits into from
Mar 29, 2024

Conversation

dbogunowicz
Copy link
Contributor

@dbogunowicz dbogunowicz commented Mar 27, 2024

A copy of this PR: #2046, but instead of introducing transformers<3.37, we install transformers==3.34.1. This is the same transformer version as in our fork.

@dbogunowicz dbogunowicz changed the base branch from main to llama_export_pwy March 28, 2024 11:21
@dbogunowicz dbogunowicz changed the base branch from llama_export_pwy to main March 28, 2024 11:21
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.

Nice job! Appreciate the modification tests and just have a few comments

src/sparseml/export/validators.py Outdated Show resolved Hide resolved
src/sparseml/transformers/utils/helpers.py Show resolved Hide resolved
symmetric: False"""


def test_modifying_llama(llama_model, shared_helper_functions):
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We should get to the point where we can test making a modified model and an unmodified model, and then run through them comparing outputs. This may help catch the forward passes changing between transformers versions. This may be too intensive for current hardware, but I think should absolutely be setup with the gpu runners - thoughts?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Good idea. This is a big step toward robust detection of problems that arise from the transformers upgrade.

setup.py Outdated Show resolved Hide resolved
mgoin
mgoin previously approved these changes Mar 29, 2024
src/sparseml/export/validators.py Outdated Show resolved Hide resolved
Co-authored-by: Michael Goin <michael@neuralmagic.com>
@mgoin mgoin merged commit 5a2970c into main Mar 29, 2024
10 of 15 checks passed
@mgoin mgoin deleted the feature/damian/forkless_4.34 branch March 29, 2024 14:33
@mgoin mgoin mentioned this pull request Apr 3, 2024
8 tasks
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.

None yet

3 participants