Constrain Qwen image test runtime

This commit is contained in:
Mikei386
2026-09-21 09:54:02 +02:00
parent fd6c57b5ed
commit aee31aa045
+11 -4
View File
@@ -723,15 +723,22 @@ services:
profiles: [qwen-image-test]
labels:
com.mike-ai.image-worker: qwen-image-2.1-test
gpus: all
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: ["${QWEN_IMAGE_21_GPU:-1}"]
capabilities: [gpu]
read_only: true
tmpfs: ["/tmp:size=4g,mode=1777"]
tmpfs:
- /tmp:size=4g,mode=1777
- /opt/ComfyUI/user:size=64m,mode=0755
- /opt/ComfyUI/temp:size=4g,mode=1777
volumes:
- "${QWEN_IMAGE_21_MODEL_DIR:-/data/models/Qwen-Image-2.1-ComfyUI}:/opt/ComfyUI/models:ro"
- "${QWEN_IMAGE_21_OUTPUT_DIR:-/data/qwen-image-2.1-test-output}:/opt/ComfyUI/output"
environment:
# Athena enumerates RTX 3060 as GPU 0 and RTX 5080 as GPU 1.
NVIDIA_VISIBLE_DEVICES: "${QWEN_IMAGE_21_GPU:-1}"
NVIDIA_DRIVER_CAPABILITIES: compute,utility
PYTORCH_CUDA_ALLOC_CONF: expandable_segments:True
command: