From 9c1fbc1da297444faa17d9b0a9e5642da68a0a5a Mon Sep 17 00:00:00 2001 From: sudolulo Date: Wed, 10 Jun 2026 15:25:57 -0400 Subject: [PATCH] Create compose.yml --- compose.yml | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 compose.yml diff --git a/compose.yml b/compose.yml new file mode 100644 index 0000000..0f2aa84 --- /dev/null +++ b/compose.yml @@ -0,0 +1,34 @@ +services: + if-curator: + build: . + image: if-curator-headless:latest + container_name: if-curator + environment: + - IMMICH_URL=${IMMICH_URL} + - API_KEY=${API_KEY} + - AUTO_MODE=true + - FRIGATE_URL=${FRIGATE_URL} + - TRAINING_MODE=face + - STRATEGY=auto + - FORCE_CPU=false + - ENABLE_CACHE=true + - CACHE_DIR=/app/.if_cache + - HF_HOME=/models/huggingface + - INSIGHTFACE_HOME=/models/insightface + # - ONLY_PEOPLE=John,Jane + # - SKIP_PEOPLE=Unknown + # - MIN_FACE_COUNT=5 + volumes: + - /mnt//if-curator/models:/models + - /mnt//if-curator/embeddings:/app/.if_cache + - /mnt//if-curator/output:/app/output + stdin_open: true + tty: true + restart: "no" + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: all + capabilities: [gpu]