File size: 1,946 Bytes
b385912 5375c2e b385912 5375c2e b385912 5375c2e b385912 | 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 | # Local-dev convenience. HF Spaces ignores this file and consumes only the
# Dockerfile. Run `docker compose up --build` to reproduce the production
# container locally on http://localhost:7860.
services:
website:
build:
context: .
dockerfile: Dockerfile
image: bani57-website:local
ports:
- "7860:7860"
# Larger /dev/shm than Docker's 64 MB default — torch's share_memory()
# falls back to it for tensor storage. The registry monkey-patches the
# COINs share_memory call to a no-op anyway, but bumping shm_size keeps
# any other PyTorch shared-memory code path from blowing up locally.
shm_size: "2gb"
environment:
DJANGO_DEBUG: "False"
# Required. Generate locally with:
# python -c "import secrets; print(secrets.token_urlsafe(50))"
# then put it in .env at the repo root (see .env.example).
DJANGO_SECRET_KEY: ${DJANGO_SECRET_KEY:?DJANGO_SECRET_KEY must be set in .env or your shell}
DJANGO_ALLOWED_HOSTS: ${DJANGO_ALLOWED_HOSTS:-localhost,127.0.0.1}
CORS_ALLOWED_ORIGINS: ${CORS_ALLOWED_ORIGINS:-http://localhost:7860}
HF_CHECKPOINTS_REPO: ${HF_CHECKPOINTS_REPO:-Bani57/checkpoints}
# Only needed for a private checkpoint repo. An empty value is safe —
# entrypoint.sh unsets HF_TOKEN when empty so huggingface_hub doesn't
# build a malformed 'Bearer ' header.
HF_TOKEN: ${HF_TOKEN:-}
# CHECKPOINTS_ROOT defaults to RESEARCH_ROOT (/app/research) in the
# image — snapshot_download writes weights alongside the bundled
# configs and Loader caches so every research-code path resolves
# against a single tree. No named volume because mixing
# image-baked files with a fresh volume mount hides the bundled
# caches and configs. ~5.4 GB is re-pulled on each `compose up`,
# which takes ~3 min with hf_transfer enabled.
TORCH_DEVICE: ${TORCH_DEVICE:-cpu}
|