From d47d214a3becbbdf542e6f91944e7978e6b8c222 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Mon, 6 May 2024 09:57:32 +0200 Subject: [PATCH] [push-ci-image] --- docker/torch-light.dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/torch-light.dockerfile b/docker/torch-light.dockerfile index c077aac839cf30..2172d66ca8a769 100644 --- a/docker/torch-light.dockerfile +++ b/docker/torch-light.dockerfile @@ -6,5 +6,5 @@ ENV VIRTUAL_ENV=/usr/local RUN pip --no-cache-dir install uv && uv venv && uv pip install --no-cache-dir -U pip setuptools RUN pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url https://download.pytorch.org/whl/cpu RUN uv pip install --no-deps timm accelerate --extra-index-url https://download.pytorch.org/whl/cpu -RUN uv pip install --no-cache-dir librosa "transformers[sklearn,sentencepiece,vision,testing" +RUN uv pip install --no-cache-dir librosa "transformers[sklearn,sentencepiece,vision,testing]" RUN pip uninstall -y transformers \ No newline at end of file