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

add sparseGPT pruner feature, refactor pruning class #1111

Merged
merged 9 commits into from
Jul 31, 2023

Conversation

WeiweiZhang1
Copy link
Contributor

Type of Change

feature
API not changed

Description

refactor sparseGPT and retrain-free, add two new post-training pruning class.

Signed-off-by: Zhang, Weiwei1 <[email protected]>
Signed-off-by: Zhang, Weiwei1 <[email protected]>
Signed-off-by: Zhang, Weiwei1 <[email protected]>
@WeiweiZhang1 WeiweiZhang1 requested a review from n1ck-guo July 25, 2023 00:43
Copy link
Contributor

@wenhuach21 wenhuach21 left a comment

Choose a reason for hiding this comment

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

The current code is not suitable to support prepare_comperssion.

Signed-off-by: Zhang, Weiwei1 <[email protected]>
@wenhuach21 wenhuach21 added the WIP label Jul 28, 2023
Signed-off-by: Zhang, Weiwei1 <[email protected]>
Signed-off-by: Zhang, Weiwei1 <[email protected]>
Signed-off-by: Zhang, Weiwei1 <[email protected]>
@chensuyue
Copy link
Contributor

We can merge this PR first, please submit another PR to update docs including 1. example readme,2. example list, 3. pruning feature docs.

@chensuyue chensuyue removed the WIP label Jul 31, 2023
@chensuyue chensuyue merged commit 88adfc9 into master Jul 31, 2023
@chensuyue chensuyue deleted the pruning/sparse_gpt_pruning branch July 31, 2023 08:00
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants