From ce5db5caaf3617f06516c8cfbfc7dc0b14d45693 Mon Sep 17 00:00:00 2001 From: Bryan Van Deusen Date: Tue, 30 Jun 2026 15:28:50 -0400 Subject: [PATCH] chore(agent): default the anatomy + panel proposer weights to working values booru_yolo yolov11m_aa22.pt (40MB) + mosesb/best-comic-panel-detection::best.pt. Each self-disables if its download fails, so defaulting them on is safe. Co-Authored-By: Claude Opus 4.8 Claude-Session: https://claude.ai/code/session_01Ttrj5P7upUTueSfoJcxEqa --- agent/docker-compose.yml | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/agent/docker-compose.yml b/agent/docker-compose.yml index d37e214..5032669 100644 --- a/agent/docker-compose.yml +++ b/agent/docker-compose.yml @@ -37,17 +37,18 @@ services: # Crop embedder (SigLIP concept bag): float16 keeps VRAM low on a shared # desktop GPU; the model itself is announced by the server. SIGLIP_DTYPE: ${SIGLIP_DTYPE:-float16} - # Crop PROPOSERS (extra YOLO detectors → more/better concept crops). + # Crop PROPOSERS (extra YOLO detectors → more/better concept crops). Each + # downloads its weights once (cached on the models volume) and self-disables + # if the download/load fails. Blank any one to turn it off. # PERSON_WEIGHTS: general COCO person detector (Western/realistic figures), - # merged with the anime person detector. Default works out of the box. - # ANATOMY_WEIGHTS: booru_yolo (anime/furry/NSFW components). Set to the - # weights URL, e.g. the yolov11m_aa22.pt from github.com/aperveyev/booru_yolo. - # PANEL_WEIGHTS: comic-panel detector as "hf_repo::file", - # e.g. mosesb/best-comic-panel-detection::.pt - # Empty = that proposer is off. Each self-disables if its weights fail to load. + # merged with the anime detector. yolo11n.pt (~6 MB, auto-downloaded). + # ANATOMY_WEIGHTS: booru_yolo anime/furry/NSFW components (~40 MB). NB the + # repo states no license — fine for private use. yolov8n_as01.pt is the + # 6 MB nano if you want lighter than yolov11m_aa22.pt. + # PANEL_WEIGHTS: mosesb comic-panel detector (Apache-2.0), "hf_repo::file". PERSON_WEIGHTS: ${PERSON_WEIGHTS:-yolo11n.pt} - ANATOMY_WEIGHTS: ${ANATOMY_WEIGHTS:-} - PANEL_WEIGHTS: ${PANEL_WEIGHTS:-} + ANATOMY_WEIGHTS: ${ANATOMY_WEIGHTS:-https://github.com/aperveyev/booru_yolo/raw/main/models/yolov11m_aa22.pt} + PANEL_WEIGHTS: ${PANEL_WEIGHTS:-mosesb/best-comic-panel-detection::best.pt} volumes: # Persist the downloaded ONNX models so restarts are fast. - fc-agent-models:/models