feat: no huggingface option

adetailer/common.py

@@ -20,7 +20,9 @@ class PredictOutput:
     preview: Optional[Image.Image] = None
 
 
-def get_models(model_dir: Union[str, Path]) -> OrderedDict[str, Optional[str]]:
+def get_models(
+    model_dir: Union[str, Path], huggingface: bool = True
+) -> OrderedDict[str, Optional[str]]:
     model_dir = Path(model_dir)
     if model_dir.is_dir():
         model_paths = [
@@ -31,17 +33,29 @@ def get_models(model_dir: Union[str, Path]) -> OrderedDict[str, Optional[str]]:
     else:
         model_paths = []
 
-    models = OrderedDict(
-        {
-            "face_yolov8n.pt": hf_hub_download(repo_id, "face_yolov8n.pt"),
-            "face_yolov8s.pt": hf_hub_download(repo_id, "face_yolov8s.pt"),
-            "mediapipe_face_full": None,
-            "mediapipe_face_short": None,
-            "hand_yolov8n.pt": hf_hub_download(repo_id, "hand_yolov8n.pt"),
-            "person_yolov8n-seg.pt": hf_hub_download(repo_id, "person_yolov8n-seg.pt"),
-            "person_yolov8s-seg.pt": hf_hub_download(repo_id, "person_yolov8s-seg.pt"),
-        }
-    )
+    if huggingface:
+        models = OrderedDict(
+            {
+                "face_yolov8n.pt": hf_hub_download(repo_id, "face_yolov8n.pt"),
+                "face_yolov8s.pt": hf_hub_download(repo_id, "face_yolov8s.pt"),
+                "mediapipe_face_full": None,
+                "mediapipe_face_short": None,
+                "hand_yolov8n.pt": hf_hub_download(repo_id, "hand_yolov8n.pt"),
+                "person_yolov8n-seg.pt": hf_hub_download(
+                    repo_id, "person_yolov8n-seg.pt"
+                ),
+                "person_yolov8s-seg.pt": hf_hub_download(
+                    repo_id, "person_yolov8s-seg.pt"
+                ),
+            }
+        )
+    else:
+        models = OrderedDict(
+            {
+                "mediapipe_face_full": None,
+                "mediapipe_face_short": None,
+            }
+        )
 
     for path in model_paths:
         if path.name in models:

preload.py

@@ -0,0 +1,9 @@
+import argparse
+
+
+def preload(parser: argparse.ArgumentParser):
+    parser.add_argument(
+        "--ad-no-huggingface",
+        action="store_true",
+        help="Don't use adetailer models from huggingface",
+    )

scripts/!adetailer.py

@@ -44,8 +44,9 @@ except Exception:
     pass
 
 AFTER_DETAILER = "After Detailer"
+no_huggingface = getattr(cmd_opts, "ad_no_huggingface", False)
 adetailer_dir = Path(models_path, "adetailer")
-model_mapping = get_models(adetailer_dir)
+model_mapping = get_models(adetailer_dir, huggingface=not no_huggingface)
 txt2img_submit_button = img2img_submit_button = None