mirror of
https://github.com/denizsafak/abogen.git
synced 2026-07-18 05:40:26 +02:00
46 lines
1.6 KiB
YAML
46 lines
1.6 KiB
YAML
services:
|
|
abogen:
|
|
build:
|
|
context: .
|
|
dockerfile: abogen/Dockerfile
|
|
args:
|
|
TORCH_INDEX_URL: ${TORCH_INDEX_URL:-https://download.pytorch.org/whl/cu124}
|
|
TORCH_VERSION: ${TORCH_VERSION:-}
|
|
image: abogen:latest
|
|
user: "${ABOGEN_UID:-1000}:${ABOGEN_GID:-1000}"
|
|
ports:
|
|
- "${ABOGEN_PORT:-8808}:8808"
|
|
volumes:
|
|
- ${ABOGEN_DATA:-./data}:/data
|
|
- ${ABOGEN_SETTINGS_DIR:-./config}:/config
|
|
- ${ABOGEN_OUTPUT_DIR:-./storage/output}:/data/outputs
|
|
- ${ABOGEN_TEMP_DIR:-./storage/tmp}:/data/cache
|
|
environment:
|
|
ABOGEN_HOST: 0.0.0.0
|
|
ABOGEN_PORT: 8808
|
|
ABOGEN_SETTINGS_DIR: "/config"
|
|
ABOGEN_UPLOAD_ROOT: /data/uploads
|
|
ABOGEN_OUTPUT_DIR: "/data/outputs"
|
|
ABOGEN_OUTPUT_ROOT: "/data/outputs"
|
|
ABOGEN_TEMP_DIR: "/data/cache"
|
|
HOME: "/data"
|
|
XDG_CACHE_HOME: "/data/cache"
|
|
HF_HOME: "/data/cache/huggingface"
|
|
HUGGINGFACE_HUB_CACHE: "/data/cache/huggingface"
|
|
TRANSFORMERS_CACHE: "/data/cache/huggingface"
|
|
# --- GPU support -----------------------------------------------------
|
|
# These settings assume the NVIDIA Container Toolkit is installed.
|
|
# Leave them in place for GPU acceleration; comment out the entire block
|
|
# below if you are deploying to a CPU-only host.
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- capabilities: [gpu]
|
|
# driver: nvidia
|
|
# count: all
|
|
# Runtime flag is only honored by legacy docker-compose (v1) CLI.
|
|
# Uncomment if you're still using it:
|
|
# runtime: nvidia
|
|
restart: unless-stopped
|