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

[Wanda Refactor] Define LayerCompressor Contract #1886

Closed
wants to merge 8 commits into from

Conversation

rahul-tuli
Copy link
Member

This PR defines a LayerCompressor contract; similar to #1885 to remove some code redundancy, and is the second piece pertaining to WandaRefactor; the last remaining piece is to Update Wanda and OBCQ modifier to share code.

@rahul-tuli rahul-tuli changed the title Define LayerCompressor Contract [Wanda Refactor] Define LayerCompressor Contract Dec 7, 2023
@rahul-tuli rahul-tuli self-assigned this Dec 7, 2023
@rahul-tuli rahul-tuli marked this pull request as ready for review December 7, 2023 16:30
src/sparseml/modifiers/utils/layer_compressors.py Outdated Show resolved Hide resolved
@rahul-tuli rahul-tuli mentioned this pull request Dec 7, 2023
3 tasks
Copy link
Member

@bfineran bfineran left a comment

Choose a reason for hiding this comment

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

LGTM pending name update

rahul-tuli added a commit that referenced this pull request Dec 18, 2023
@rahul-tuli
Copy link
Member Author

Closing cause requested changes are now a part of #1887

@rahul-tuli rahul-tuli closed this Dec 18, 2023
rahul-tuli added a commit that referenced this pull request Dec 18, 2023
* Define GPT contract

* rename tmp -> batch_size

* Define LayerCompressor Contract

* Rename gpt_helpers to gpts
Fix some docstrings

* add named argument to function call

* Wanda/OBCQ refactor

* propagate target-ids

* Address review comments from
* #1885
* #1886
@rahul-tuli rahul-tuli deleted the layer-compressor-contract branch December 19, 2023 21:11
rahul-tuli added a commit that referenced this pull request Dec 28, 2023
* Add wanda base

* Initial implementation

* Update Wanda Base

* Refactor to use WandaLayerCompressor
Update WrappedGPT

* Rename WrappedGPT to WandaGPT

* [Wanda Refactor] Wanda/OBCQ Modifier Refactor (#1887)

* Define GPT contract

* rename tmp -> batch_size

* Define LayerCompressor Contract

* Rename gpt_helpers to gpts
Fix some docstrings

* add named argument to function call

* Wanda/OBCQ refactor

* propagate target-ids

* Address review comments from
* #1885
* #1886

* Fix typo

* Update test

* Fix regression
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants