add download command

lama_cleaner/diffusers_utils.py

@@ -0,0 +1,28 @@
+import json
+from pathlib import Path
+from typing import Dict, List
+
+
+def folder_name_to_show_name(name: str) -> str:
+    return name.replace("models--", "").replace("--", "/")
+
+
+def _scan_models(cache_dir, class_name: str) -> List[str]:
+    cache_dir = Path(cache_dir)
+    res = []
+    for it in cache_dir.glob("**/*/model_index.json"):
+        with open(it, "r", encoding="utf-8") as f:
+            data = json.load(f)
+            if data["_class_name"] == class_name:
+                name = folder_name_to_show_name(it.parent.parent.parent.name)
+                if name not in res:
+                    res.append(name)
+    return res
+
+
+def scan_models(cache_dir) -> List[str]:
+    return _scan_models(cache_dir, "StableDiffusionPipeline")
+
+
+def scan_inpainting_models(cache_dir) -> List[str]:
+    return _scan_models(cache_dir, "StableDiffusionInpaintPipeline")