Add venv support to FLUX worker image

2026-08-22 17:45:50 +02:00
parent facc224c1b
commit 364ac6b83e
1 changed files with 3 additions and 1 deletions
@@ -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}" \