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

* fix failing 8bit test * trigger CI
blbadger · Nov 8, 2023 · eb2e114 · eb2e114
1 parent 6851e4b
commit eb2e114
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