From 8379467e13480e6c8c95c0b783b0ce0ae35f12d8 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Sun, 28 Apr 2024 15:15:37 +0200 Subject: [PATCH] [push-ci-image] --- docker/consistency.dockerfile | 4 +++- docker/torch-light.dockerfile | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/docker/consistency.dockerfile b/docker/consistency.dockerfile index f1b7d34ce0e904..fbab8477e85989 100644 --- a/docker/consistency.dockerfile +++ b/docker/consistency.dockerfile @@ -11,4 +11,6 @@ RUN uv pip install --no-cache --upgrade 'torch' --index-url https://download.pyt RUN uv pip install --no-cache-dir --upgrade GitPython RUN uv pip install --no-cache-dir "transformers[flax,quality,vision]" RUN pip uninstall -y transformers -RUN apt-get clean && rm -rf /var/lib/apt/lists/* \ No newline at end of file +RUN apt-get clean && rm -rf /var/lib/apt/lists/* + + diff --git a/docker/torch-light.dockerfile b/docker/torch-light.dockerfile index f2ba6e01de281e..8a4b07340c609d 100644 --- a/docker/torch-light.dockerfile +++ b/docker/torch-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 "pytest<8.0.1" "fsspec>=2023.5.0,<2023.10.0" pytest-subtests pytest-xdist RUN 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,vision,timm]" +RUN uv pip install --no-cache-dir "transformers[sklearn,sentencepiece,vision,timm]" RUN pip uninstall -y transformers