From b95e1affc15b7c895cf6b4b48f3ee40c191db150 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Thu, 2 May 2024 09:19:22 +0200 Subject: [PATCH] [push-ci-image] --- docker/tf-light.dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docker/tf-light.dockerfile b/docker/tf-light.dockerfile index d3ede63f8c497b..91487e9a5ef7c4 100644 --- a/docker/tf-light.dockerfile +++ b/docker/tf-light.dockerfile @@ -5,7 +5,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends libsndfile1-de RUN apt-get install -y cmake ENV VIRTUAL_ENV=/usr/local RUN pip --no-cache-dir install uv && uv venv && uv pip install --no-cache-dir -U pip setuptools -RUN uv pip install --upgrade --no-cache-dir "transformers[sklearn,tf-cpu,testing,sentencepiece,tf-speech,vision]" +RUN uv pip install "keras<2.16" +RUN uv pip install --upgrade --no-cache-dir "transformers[tf-cpu,sklearn,testing,sentencepiece,tf-speech,vision]" RUN uv pip install --no-cache-dir "protobuf==3.20.3" tensorflow_probability RUN pip uninstall -y transformers RUN apt-get clean && rm -rf /var/lib/apt/lists/* && apt-get autoremove && apt-get autoclean \ No newline at end of file