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] Wanda/OBCQ Modifier Refactor #1887

Merged
merged 8 commits into from
Dec 18, 2023

Conversation

rahul-tuli
Copy link
Member

Wanda can be thought of as a simpler/more-general case of OBCQ; this PR leverages that to reduce code duplicacy

@rahul-tuli rahul-tuli marked this pull request as ready for review December 7, 2023 18:22
@rahul-tuli rahul-tuli self-assigned this Dec 7, 2023
@rahul-tuli rahul-tuli mentioned this pull request Dec 7, 2023
3 tasks
@rahul-tuli rahul-tuli changed the base branch from layer-compressor-contract to wanda December 18, 2023 14:53
@rahul-tuli rahul-tuli merged commit 5f24ff9 into wanda Dec 18, 2023
@rahul-tuli rahul-tuli deleted the wanda-modifier-refactor branch December 18, 2023 14:55
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

1 participant