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: