michael/AI-Profile-Router
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Constrain Qwen image test runtime

Browse Source
This commit is contained in:
Mikei386
2026-09-21 09:54:02 +02:00
parent fd6c57b5ed
commit aee31aa045
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compose.yaml
+11 -4
View File
@@ -723,15 +723,22 @@ services:
profiles: [qwen-image-test] profiles: [qwen-image-test]
labels: labels:
com.mike-ai.image-worker: qwen-image-2.1-test 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 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: volumes:
- "${QWEN_IMAGE_21_MODEL_DIR:-/data/models/Qwen-Image-2.1-ComfyUI}:/opt/ComfyUI/models:ro" - "${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" - "${QWEN_IMAGE_21_OUTPUT_DIR:-/data/qwen-image-2.1-test-output}:/opt/ComfyUI/output"
environment: 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 NVIDIA_DRIVER_CAPABILITIES: compute,utility
PYTORCH_CUDA_ALLOC_CONF: expandable_segments:True PYTORCH_CUDA_ALLOC_CONF: expandable_segments:True
command: command: