65 lines
1.6 KiB
YAML
65 lines
1.6 KiB
YAML
|
# Docker Compose for local development and testing
|
||
|
|
# For RunPod deployment, use the Dockerfile directly
|
||
|
|
|
||
|
|
version: '3.8'
|
||
|
|
|
||
|
|
services:
|
||
|
|
audiocraft:
|
||
|
|
build:
|
||
|
|
context: .
|
||
|
|
dockerfile: Dockerfile
|
||
|
|
container_name: audiocraft-studio
|
||
|
|
ports:
|
||
|
|
- "7860:7860" # Gradio UI
|
||
|
|
- "8000:8000" # REST API
|
||
|
|
volumes:
|
||
|
|
# Persistent storage
|
||
|
|
- audiocraft-outputs:/workspace/outputs
|
||
|
|
- audiocraft-data:/workspace/data
|
||
|
|
- audiocraft-models:/workspace/models
|
||
|
|
# Development: mount source code
|
||
|
|
- ./src:/workspace/app/src:ro
|
||
|
|
- ./config:/workspace/app/config:ro
|
||
|
|
environment:
|
||
|
|
- AUDIOCRAFT_HOST=0.0.0.0
|
||
|
|
- AUDIOCRAFT_GRADIO_PORT=7860
|
||
|
|
- AUDIOCRAFT_API_PORT=8000
|
||
|
|
- AUDIOCRAFT_DEBUG=false
|
||
|
|
- AUDIOCRAFT_COMFYUI_RESERVE_GB=0 # No ComfyUI in this compose
|
||
|
|
- NVIDIA_VISIBLE_DEVICES=all
|
||
|
|
deploy:
|
||
|
|
resources:
|
||
|
|
reservations:
|
||
|
|
devices:
|
||
|
|
- driver: nvidia
|
||
|
|
count: 1
|
||
|
|
capabilities: [gpu]
|
||
|
|
restart: unless-stopped
|
||
|
|
healthcheck:
|
||
|
|
test: ["CMD", "curl", "-f", "http://localhost:7860/"]
|
||
|
|
interval: 30s
|
||
|
|
timeout: 10s
|
||
|
|
retries: 3
|
||
|
|
start_period: 60s
|
||
|
|
|
||
|
|
# Optional: Run alongside ComfyUI
|
||
|
|
# comfyui:
|
||
|
|
# image: your-comfyui-image
|
||
|
|
# container_name: comfyui
|
||
|
|
# ports:
|
||
|
|
# - "8188:8188"
|
||
|
|
# volumes:
|
||
|
|
# - comfyui-data:/workspace
|
||
|
|
# deploy:
|
||
|
|
# resources:
|
||
|
|
# reservations:
|
||
|
|
# devices:
|
||
|
|
# - driver: nvidia
|
||
|
|
# count: 1
|
||
|
|
# capabilities: [gpu]
|
||
|
|
|
||
|
|
volumes:
|
||
|
|
audiocraft-outputs:
|
||
|
|
audiocraft-data:
|
||
|
|
audiocraft-models:
|