Spaces:
Running on Zero
Running on Zero
File size: 1,391 Bytes
b701455 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | services:
lightdiffusion:
build:
context: .
dockerfile: Dockerfile
args:
# Specify target GPU architectures for CUDA extension builds
# 8.0: A100, 8.6: RTX 30xx, 8.9: RTX 40xx, 9.0: H100, 12.0: RTX 50xx (Blackwell)
# Customize based on your GPU: TORCH_CUDA_ARCH_LIST: "12.0" for RTX 50xx only
TORCH_CUDA_ARCH_LIST: "8.0;8.6;8.9;9.0;12.0"
INSTALL_STABLE_FAST: "0"
INSTALL_OLLAMA: "0"
INSTALL_SAGEATTENTION: "0"
INSTALL_SPARGEATTN: "0"
ports:
- "7860:7860" # FastAPI backend serving the built React UI
volumes:
# Mount output directory to persist generated images
- ./output:/app/output
# Mount checkpoints directory for model files
- ./include/checkpoints:/app/include/checkpoints
# Mount other model directories
- ./include/loras:/app/include/loras
- ./include/embeddings:/app/include/embeddings
- ./include/ESRGAN:/app/include/ESRGAN
- ./include/yolos:/app/include/yolos
environment:
- PORT=7860
- CUDA_VISIBLE_DEVICES=0
- CUDA_HOME=/usr/local/cuda
- PROMPT_ENHANCER_MODEL=qwen3:0.6b
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [ gpu ]
restart: unless-stopped
stdin_open: true
tty: true
|