Skip to content

GPTQ quantizer and example of using it to quantize chatglm-6b#1100

Open
chenbohua3 wants to merge 8 commits intoalibaba:mainfrom
chenbohua3:gptq_quantizer
Open

GPTQ quantizer and example of using it to quantize chatglm-6b#1100
chenbohua3 wants to merge 8 commits intoalibaba:mainfrom
chenbohua3:gptq_quantizer

Conversation

@chenbohua3
Copy link
Collaborator

No description provided.

model = MyModel()
ss = dict(model.named_modules())
quantizer = GPTQuantizer()
calib_model = quantizer.calib(model)
Copy link
Collaborator

Choose a reason for hiding this comment

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

calib_model,相对于model加了一个hook,calib_model is model是True吧?

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

是的,只是为了保持接口和常规量化一致

@chenbohua3 chenbohua3 changed the title [WIP] A simplified version of GPTQ quantizer GPTQ quantizer and example of chatglm-6b Apr 12, 2023
@chenbohua3 chenbohua3 changed the title GPTQ quantizer and example of chatglm-6b GPTQ quantizer and example of using it to quantize chatglm-6b Apr 12, 2023
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.

2 participants

Comments