From 727e9b6d396548d0efcd0b02ed9386cce1b91b59 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Thu, 2 May 2024 09:28:48 +0200 Subject: [PATCH] [push-ci-image] --- docker/pipeline-tf.dockerfile | 2 +- docker/torch-tf-light.dockerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/pipeline-tf.dockerfile b/docker/pipeline-tf.dockerfile index c227bc9c159796..bd46d17a1a60df 100644 --- a/docker/pipeline-tf.dockerfile +++ b/docker/pipeline-tf.dockerfile @@ -3,7 +3,7 @@ ENV PYTHONDONTWRITEBYTECODE=1 USER root RUN apt-get update && apt-get install -y libsndfile1-dev espeak-ng time git cmake g++ ENV VIRTUAL_ENV=/usr/local -RUN pip --no-cache-dir install uv && uv venv && uv pip install --no-cache-dir -U pip setuptools +RUN pip --no-cache-dir install uv && uv venv && uv pip install --no-cache-dir -U pip setuptools "keras<2.16" RUN uv pip install --no-cache-dir "transformers[sklearn,tf-cpu,testing,sentencepiece,tf-speech,vision]" RUN uv pip install --no-cache-dir "protobuf==3.20.3" tensorflow_probability RUN apt-get clean && rm -rf /var/lib/apt/lists/* \ No newline at end of file diff --git a/docker/torch-tf-light.dockerfile b/docker/torch-tf-light.dockerfile index 2d0617ea43a5fc..2907b989e5f335 100644 --- a/docker/torch-tf-light.dockerfile +++ b/docker/torch-tf-light.dockerfile @@ -8,7 +8,7 @@ RUN uv pip install --no-cache-dir --no-deps accelerate --extra-index-url https: RUN pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url https://download.pytorch.org/whl/cpu RUN git lfs install -RUN uv pip install --no-cache-dir pypi-kenlm +RUN uv pip install --no-cache-dir pypi-kenlm "keras<2.16" RUN uv pip install --no-cache-dir "transformers[tf-cpu,sklearn,sentencepiece,vision, testing]" RUN uv pip install --no-cache-dir "protobuf==3.20.3" tensorflow_probability