From bd163142777b495f90bc3a0540dff4b18f993bb3 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Fri, 26 Apr 2024 18:49:56 +0200 Subject: [PATCH] [push-ci-image] --- docker/torch-jax-light.dockerfile | 2 +- docker/torch-light.dockerfile | 2 +- docker/torch-tf-light.dockerfile | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/torch-jax-light.dockerfile b/docker/torch-jax-light.dockerfile index 502cb210ae9668..a81d3dc0e8a19c 100644 --- a/docker/torch-jax-light.dockerfile +++ b/docker/torch-jax-light.dockerfile @@ -9,7 +9,7 @@ RUN uv venv RUN uv pip install --no-cache-dir -U pip setuptools accelerate RUN uv pip install --no-cache-dir "pytest<8.0.1" "fsspec>=2023.5.0,<2023.10.0" pytest-subtests pytest-xdist -RUN uv pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu +RUN uv pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url https://download.pytorch.org/whl/cpu RUN uv pip install --no-cache-dir "transformers[sklearn,flax,torch,testing,sentencepiece,torch-speech,vision]" diff --git a/docker/torch-light.dockerfile b/docker/torch-light.dockerfile index 68f0c7e117df2b..6a5f69600cc848 100644 --- a/docker/torch-light.dockerfile +++ b/docker/torch-light.dockerfile @@ -8,7 +8,7 @@ RUN pip --no-cache-dir install uv RUN uv venv RUN uv pip install --no-cache-dir -U pip setuptools RUN uv pip install --no-cache-dir "pytest<8.0.1" "fsspec>=2023.5.0,<2023.10.0" pytest-subtests pytest-xdist -RUN uv pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu +RUN uv pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url https://download.pytorch.org/whl/cpu RUN uv pip install --no-cache-dir "transformers[sklearn,torch,testing,sentencepiece,torch-speech,vision,timm]" diff --git a/docker/torch-tf-light.dockerfile b/docker/torch-tf-light.dockerfile index 86edc2f3a144c1..6acf843e7be12e 100644 --- a/docker/torch-tf-light.dockerfile +++ b/docker/torch-tf-light.dockerfile @@ -9,7 +9,7 @@ RUN uv venv RUN uv pip install --no-cache-dir -U pip setuptools -RUN uv pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu +RUN uv pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url https://download.pytorch.org/whl/cpu RUN uv pip install --no-cache-dir pypi-kenlm RUN uv pip install --no-cache-dir "transformers[sklearn,tf-cpu,torch,testing,sentencepiece,torch-speech,vision]" RUN uv pip install --no-cache-dir -U tensorflow_probability accelerate