Skip to content

GitLab

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

FIX: re-add bnb on docker image (#30427) 

Update Dockerfile
parent 4b63d013
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
docker/transformers-all-latest-gpu/Dockerfile
View file @ d179b9dc
...@@ -45,6 +45,9 @@ RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/opt ...@@ -45,6 +45,9 @@ RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/opt
# For video model testing # For video model testing
RUN python3 -m pip install --no-cache-dir decord av==9.2.0 RUN python3 -m pip install --no-cache-dir decord av==9.2.0
# Some slow tests require bnb
RUN python3 -m pip install --no-cache-dir bitsandbytes
# For `dinat` model # For `dinat` model
# The `XXX` part in `torchXXX` needs to match `PYTORCH` (to some extent) # The `XXX` part in `torchXXX` needs to match `PYTORCH` (to some extent)
RUN python3 -m pip install --no-cache-dir natten==0.15.1+torch220$CUDA -f https://shi-labs.com/natten/wheels RUN python3 -m pip install --no-cache-dir natten==0.15.1+torch220$CUDA -f https://shi-labs.com/natten/wheels
......
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