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

[Release 0.3.2] Additional patches to enable compatibility with SparseML, UX changes #43

Merged
merged 5 commits into from
Apr 29, 2024

Conversation

Satrat
Copy link

@Satrat Satrat commented Apr 26, 2024

Changing the helper infer_compressor_from_model_config to ModelCompressor.from_pretrained to match more closely with QuantizationConfig.from_pretrained

Also adding a device default to decompress to fix some errors in the sparseml tests

bfineran
bfineran previously approved these changes Apr 26, 2024
@dbogunowicz
Copy link
Contributor

@Satrat sweet. Let's get this in as well as #41 and then push the new pypi package.

dbogunowicz
dbogunowicz previously approved these changes Apr 29, 2024
@dbogunowicz dbogunowicz dismissed stale reviews from bfineran and themself via 538d587 April 29, 2024 09:34
@dbogunowicz
Copy link
Contributor

Or better yet, I merged those two PRs.

@dbogunowicz dbogunowicz changed the title Minor UX Changes [Release 0.3.2] Additional patches to enable compatibility with SparseML, UX changes Apr 29, 2024
@Satrat Satrat requested a review from dbogunowicz April 29, 2024 13:54
@dbogunowicz dbogunowicz merged commit 0a981af into main Apr 29, 2024
2 checks passed
@dbogunowicz dbogunowicz deleted the compressor_ux branch April 29, 2024 13:56
@dbogunowicz
Copy link
Contributor

@bfineran @Satrat landed. will build the wheel and ping DanH.

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