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