Skip to content

GitLab

Unverified Commit d4c0aa14 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[`Tests`] Fix failing 8bit test (#25564) 

* fix failing 8bit test

* trigger CI
parent 181d778f
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
docker/transformers-all-latest-gpu/Dockerfile
View file @ d4c0aa14
...@@ -50,6 +50,9 @@ RUN python3 -m pip install --no-cache-dir bitsandbytes ...@@ -50,6 +50,9 @@ RUN python3 -m pip install --no-cache-dir bitsandbytes
# Add auto-gptq for gtpq quantization testing # Add auto-gptq for gtpq quantization testing
RUN python3 -m pip install --no-cache-dir auto-gptq RUN python3 -m pip install --no-cache-dir auto-gptq
# Add einops for additional model testing
RUN python3 -m pip install --no-cache-dir einops
# For bettertransformer + gptq # For bettertransformer + gptq
RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/optimum@main#egg=optimum RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/optimum@main#egg=optimum
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment