Skip to content

Commit 4b8d1f7

Browse files
authored
Fix : adding einops lib in the CI docker for some bitsandbytes tests (#35652)
* fix docker * fix
1 parent 34f76bb commit 4b8d1f7

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

docker/transformers-all-latest-gpu/Dockerfile

+3
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,9 @@ RUN python3 -m pip install --no-cache-dir python-Levenshtein
6565
# For `FastSpeech2ConformerTokenizer` tokenizer
6666
RUN python3 -m pip install --no-cache-dir g2p-en
6767

68+
# For Some bitsandbytes tests
69+
RUN python3 -m pip install --no-cache-dir einops
70+
6871
# When installing in editable mode, `transformers` is not recognized as a package.
6972
# this line must be added in order for python to be aware of transformers.
7073
RUN cd transformers && python3 setup.py develop

0 commit comments

Comments
 (0)