[push-ci-image]

huggingface · May 2, 2024 · 9166aec · 9166aec
1 parent 47f52c1
commit 9166aec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/torch-tf-light.dockerfile b/docker/torch-tf-light.dockerfile
@@ -9,7 +9,7 @@ RUN pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url ht
 RUN git lfs install
 
 RUN uv pip install --no-cache-dir pypi-kenlm
-RUN pip install --no-cache-dir "transformers[tf-cpu,sklearn,sentencepiece,vision, testing]"
+RUN pip install --no-cache-dir "transformers[tf-cpu,sklearn,sentencepiece,vision, testing]" tensorflow-probability
 RUN uv pip install --no-cache-dir  "protobuf==3.20.3" librosa