From 9166aec31acdbf13759c42feb68aea2c36bc0db4 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Thu, 2 May 2024 14:24:35 +0200 Subject: [PATCH] [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 7e4f90f5c2e4a1..b9461fd29871a0 100644 --- 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