add get_diffusers_models

2023-11-16 21:45:55 +08:00
parent 1d145d1cd6
commit ef1179a858
2 changed files with 30 additions and 8 deletions
--- a/lama_cleaner/server.py
+++ b/lama_cleaner/server.py
@@ -2,6 +2,8 @@
 import os
 import hashlib

+from lama_cleaner.diffusers_utils import scan_models
+
 os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"

 import imghdr
@@ -432,6 +434,13 @@ def get_server_config():
    }, 200


+@app.route("/sd_models", methods=["GET"])
+def get_diffusers_models():
+    from diffusers.utils import DIFFUSERS_CACHE
+
+    return scan_models(DIFFUSERS_CACHE)
+
+
@app.route("/model")
 def current_model():
    return model.name, 200