[push-ci-image]

huggingface · Apr 27, 2024 · 8b93057 · 8b93057
1 parent 60b0e7c
commit 8b93057
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,torch-speech,vision,timm]"
+RUN uv pip install --no-cache-dir "transformers[sklearn,torch,testing,sentencepiece,vision,timm]"
 
 
 RUN pip uninstall -y transformers