From f2497de213e3c4916e7e1111b5f703ff04cb9b7c Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Fri, 26 Apr 2024 18:07:44 +0200 Subject: [PATCH] nit [push-ci-image] --- docker/torch-tf-light.dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/torch-tf-light.dockerfile b/docker/torch-tf-light.dockerfile index 6e44131d4d6b60..244588de23b55b 100644 --- a/docker/torch-tf-light.dockerfile +++ b/docker/torch-tf-light.dockerfile @@ -8,7 +8,7 @@ RUN uv venv RUN uv pip install --no-cache-dir -U pip setuptools -RUN uv pip install --no-cache-dir --upgrade torch torchvision torchaudio--index-url https://download.pytorch.org/whl/cpu +RUN uv pip install --no-cache-dir --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu 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