-
Notifications
You must be signed in to change notification settings - Fork 27k
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
[AutoGPTQ
] Add correct installation of GPTQ library + fix slow tests
#25713
[AutoGPTQ
] Add correct installation of GPTQ library + fix slow tests
#25713
Conversation
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.
works for me. Thank you for confirming the build is successful.
So basically we want to install the same version but with cu118 support, right?
The documentation is not available anymore as the PR was closed or merged. |
Can confirm the slow tests now pass in the docker image
It is slightly trickier than that, auto-gptq listed in pypi does not contain the latest supported versions (>= |
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.
LGTM ! Thanks for checking that the image is built correctly :)
huggingface#25713) * add correct installation of GPTQ library * update tests values
huggingface#25713) * add correct installation of GPTQ library * update tests values
huggingface#25713) * add correct installation of GPTQ library * update tests values
What does this PR do?
Per the instructions of installing auto-gptq library we need to slightly update the Dockerfile otherwise it will install the
0.3.2
version which is not compatible with the integrationWill also update some expected values to make the slow tests pass
cc @ydshieh @SunMarc
Can confirm the Docker image is built successfully