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

* fix failing 8bit test * trigger CI
huggingface · Aug 17, 2023 · d4c0aa1 · d4c0aa1
1 parent 181d778
commit d4c0aa1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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