mirror of
https://github.com/OpenBMB/VoxCPM.git
synced 2026-09-20 10:53:32 +08:00
da5b2da097
- nginx: return 200 OK on GET / for load balancer health checks - Dockerfile: set HF_HOME=/app/models so Hub downloads persist in mounted volume - Dockerfile: add /app/data directory and volume declaration - docker-compose: explicit volume mounts for models, data, lora, output - README: document where to put training files and find output - .gitignore: exclude volume mount directories (models/, data/, lora/, output/)
46 lines
1.2 KiB
YAML
46 lines
1.2 KiB
YAML
version: "3.8"
|
|
|
|
services:
|
|
training-webui:
|
|
build:
|
|
context: ..
|
|
dockerfile: docker/Dockerfile
|
|
ports:
|
|
- "7860:7860"
|
|
volumes:
|
|
# Pretrained model weights + HF cache (HF_HOME=/app/models in Dockerfile).
|
|
# Pre-populate with model dirs, or leave empty — auto-downloads on first run.
|
|
- ../models:/app/models
|
|
|
|
# Training data: JSONL manifests and audio files.
|
|
# Reference paths inside the container as /app/data/train.jsonl etc.
|
|
- ../data:/app/data
|
|
|
|
# LoRA training output — checkpoints, configs, logs.
|
|
# Results appear in lora/<run-name>/checkpoints/ after training.
|
|
- ../lora:/app/lora
|
|
|
|
# Additional training artifacts.
|
|
- ../output:/app/output
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: 1
|
|
capabilities: [gpu]
|
|
environment:
|
|
- GRADIO_SERVER_PORT=7860
|
|
- GRADIO_ROOT_PATH=/webui # Matches nginx location block
|
|
restart: unless-stopped
|
|
|
|
nginx:
|
|
image: nginx:alpine
|
|
ports:
|
|
- "80:80"
|
|
volumes:
|
|
- ./nginx.conf:/etc/nginx/conf.d/default.conf:ro
|
|
depends_on:
|
|
- training-webui
|
|
restart: unless-stopped
|