Skip to content

GitLab

Unverified Commit 5fd5ef76 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix docker file (#28452) 



fix docker file
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent d019acb8
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
docker/transformers-all-latest-gpu/Dockerfile
View file @ 5fd5ef76
...@@ -11,7 +11,7 @@ SHELL ["sh", "-lc"] ...@@ -11,7 +11,7 @@ SHELL ["sh", "-lc"]
ARG PYTORCH='2.1.1' ARG PYTORCH='2.1.1'
# (not always a valid torch version) # (not always a valid torch version)
ARG INTEL_TORCH_EXT='2.1.1' ARG INTEL_TORCH_EXT='2.1.100'
# Example: `cu102`, `cu113`, etc. # Example: `cu102`, `cu113`, etc.
ARG CUDA='cu118' ARG CUDA='cu118'
...@@ -65,7 +65,7 @@ RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/opt ...@@ -65,7 +65,7 @@ RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/opt
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
# For `dinat` model # For `dinat` model
RUN python3 -m pip install --no-cache-dir natten<0.15.0 -f https://shi-labs.com/natten/wheels/$CUDA/ RUN python3 -m pip install --no-cache-dir 'natten<0.15.0' -f https://shi-labs.com/natten/wheels/$CUDA/
# For `nougat` tokenizer # For `nougat` tokenizer
RUN python3 -m pip install --no-cache-dir python-Levenshtein RUN python3 -m pip install --no-cache-dir python-Levenshtein
......
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