-
Notifications
You must be signed in to change notification settings - Fork 396
Deprecate TransformerEvalWrapper, LMEvalInputRecorder
#3617
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
base: main
Are you sure you want to change the base?
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/3617
Note: Links to docs will display an error until the docs builds have been completed. This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
@pytorchbot label "topic: deprecation" |
|
oh does this work with gemma3 as well? I remember I need to make some modifications before: 2a98f58 |
torchao/quantization/GPTQ/README.md
Outdated
| ``` | ||
| For using lm_eval as a calibration data source, you would: | ||
| 1. Run lm_eval on your model with calibration tasks | ||
| 2. Collect the inputs during that run using `MultiTensorInputRecorder` |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
code examples?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I didn't consider example code because: 1) not familiar with GPTQ workflow, 2) GPTQ api migration is work in progress at #3517. How about focusing on migration instead?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Updated the code example for temporary filling up, please take a look.
Not yet, but isn't it |
Summary:
Calibration-based PTQ methods like AWQ and GPTQ have been used
TransformerEvalWrapper, which wraps ao model to runlm-eval. Instead, we want to runlm-evaldirectly, by usingHFLM. The change log can be summarized toTest plan: