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

* fix failing 8bit test * trigger CI
2023-08-17 17:34:25 +02:00
parent 181d778f83
commit d4c0aa1443
1 changed files with 3 additions and 0 deletions
--- a/docker/transformers-all-latest-gpu/Dockerfile
+++ b/docker/transformers-all-latest-gpu/Dockerfile
@@ -50,6 +50,9 @@ RUN python3 -m pip install --no-cache-dir bitsandbytes
 # Add auto-gptq for gtpq quantization testing
 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 
 RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/optimum@main#egg=optimum