From 5fd5ef76243bf7e0285cc8908ff63c2c5c412199 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Thu, 11 Jan 2024 15:34:05 +0100 Subject: [PATCH] Fix docker file (#28452) fix docker file Co-authored-by: ydshieh --- docker/transformers-all-latest-gpu/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/transformers-all-latest-gpu/Dockerfile b/docker/transformers-all-latest-gpu/Dockerfile index 96525e247f..3ee774270b 100644 --- a/docker/transformers-all-latest-gpu/Dockerfile +++ b/docker/transformers-all-latest-gpu/Dockerfile @@ -11,7 +11,7 @@ SHELL ["sh", "-lc"] ARG PYTORCH='2.1.1' # (not always a valid torch version) -ARG INTEL_TORCH_EXT='2.1.1' +ARG INTEL_TORCH_EXT='2.1.100' # Example: `cu102`, `cu113`, etc. ARG CUDA='cu118' @@ -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 # 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 RUN python3 -m pip install --no-cache-dir python-Levenshtein