diff --git a/docker/consistency.dockerfile b/docker/consistency.dockerfile index 8c77d2dda4073d..3ca01b9951fdb7 100644 --- a/docker/consistency.dockerfile +++ b/docker/consistency.dockerfile @@ -8,6 +8,7 @@ RUN uv venv RUN uv pip install --no-cache-dir -U pip setuptools RUN uv pip install --no-cache-dir tensorflow-cpu tf-keras pytest RUN uv pip install --no-cache --upgrade 'torch' --index-url https://download.pytorch.org/whl/cpu -RUN uv pip install --no-cache-dir --upgrade GitPython "transformers[flax,quality]" +RUN uv pip install --no-cache-dir --upgrade GitPython +RUN uv pip install --no-cache-dir "transformers[flax,quality]" RUN pip uninstall -y transformers RUN apt-get clean && rm -rf /var/lib/apt/lists/* \ No newline at end of file