From aee31aa045904200f60a35d45a025b2d61c0a9c7 Mon Sep 17 00:00:00 2001 From: Mikei386 <44135113+Mikei386@users.noreply.github.com> Date: Mon, 21 Sep 2026 09:54:02 +0200 Subject: [PATCH] Constrain Qwen image test runtime --- compose.yaml | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/compose.yaml b/compose.yaml index f339522..92b287b 100644 --- a/compose.yaml +++ b/compose.yaml @@ -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: