Add RTX 5080 FLUX hot-swap worker

2026-08-22 17:40:32 +02:00
parent 8267a85a96
commit 7ac93befc4
12 changed files with 398 additions and 41 deletions
@@ -0,0 +1,18 @@
+FROM pytorch/pytorch:2.11.0-cuda12.8-cudnn9-runtime
+
+ARG DIFFUSERS_VERSION=0.40.0
+ARG TRANSFORMERS_VERSION=5.15.1
+ARG ACCELERATE_VERSION=1.14.0
+ARG HF_HUB_VERSION=1.28.0
+
+RUN pip install --no-cache-dir \
+      "diffusers==${DIFFUSERS_VERSION}" \
+      "transformers==${TRANSFORMERS_VERSION}" \
+      "accelerate==${ACCELERATE_VERSION}" \
+      "huggingface-hub==${HF_HUB_VERSION}" \
+      sentencepiece protobuf safetensors pillow && \
+    useradd --system --uid 10002 --home /nonexistent --shell /usr/sbin/nologin flux
+
+COPY flux_worker.py /app/flux_worker.py
+USER 10002:10002
+ENTRYPOINT ["python", "/app/flux_worker.py"]