Constrain Qwen image test runtime
This commit is contained in:
+11
-4
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user