fix kera3 issue

huggingface · Apr 25, 2024 · b225fb7 · b225fb7
1 parent a1f7caf
commit b225fb7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/consistency.dockerfile b/docker/consistency.dockerfile
@@ -6,7 +6,7 @@ ENV VIRTUAL_ENV=/usr/local
 RUN pip install uv
 RUN uv venv
 RUN uv pip install --no-cache-dir -U pip setuptools
-RUN uv pip install --no-cache-dir tensorflow-cpu
+RUN uv pip install --no-cache-dir tensorflow-cpu tf-keras
 RUN uv pip install --no-cache --upgrade 'torch' --index-url https://download.pytorch.org/whl/cpu
 RUN pip install --no-cache-dir --upgrade GitPython "transformers[flax,quality]"
 RUN pip uninstall -y transformers