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