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::<weights>.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