services: lightdiffusion: build: context: . dockerfile: Dockerfile ports: - "7860:7860" volumes: # Mount output directory to persist generated images - ./output:/app/_internal/output # Mount checkpoints directory for model files - ./checkpoints:/app/_internal/checkpoints # Mount other model directories - ./loras:/app/_internal/loras - ./embeddings:/app/_internal/embeddings - ./ESRGAN:/app/_internal/ESRGAN - ./yolos:/app/_internal/yolos environment: - GRADIO_SERVER_NAME=0.0.0.0 - GRADIO_SERVER_PORT=7860 - CUDA_VISIBLE_DEVICES=0 deploy: resources: reservations: devices: - driver: nvidia count: 1 capabilities: [gpu] restart: unless-stopped stdin_open: true tty: true