diff --git a/platform/docker/flux-worker/Dockerfile b/platform/docker/flux-worker/Dockerfile index 3ebb41b..3d44d7a 100644 --- a/platform/docker/flux-worker/Dockerfile +++ b/platform/docker/flux-worker/Dockerfile @@ -5,7 +5,9 @@ ARG TRANSFORMERS_VERSION=5.15.1 ARG ACCELERATE_VERSION=1.14.0 ARG HF_HUB_VERSION=1.28.0 -RUN python -m venv --system-site-packages /opt/flux-venv && \ +RUN apt-get update && apt-get install -y --no-install-recommends python3.12-venv && \ + rm -rf /var/lib/apt/lists/* && \ + python -m venv --system-site-packages /opt/flux-venv && \ /opt/flux-venv/bin/pip install --no-cache-dir \ "diffusers==${DIFFUSERS_VERSION}" \ "transformers==${TRANSFORMERS_VERSION}" \