From 40ddbba25c5b2448c07b4c4824eeb35d03a92474 Mon Sep 17 00:00:00 2001 From: ImDarkTom Date: Sun, 8 Sep 2024 20:18:04 +0100 Subject: [PATCH] Add list models route --- server.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/server.py b/server.py index 3611880da..0c4a77179 100644 --- a/server.py +++ b/server.py @@ -155,6 +155,10 @@ class PromptServer(): def get_embeddings(self): embeddings = folder_paths.get_filename_list("embeddings") return web.json_response(list(map(lambda a: os.path.splitext(a)[0], embeddings))) + + @routes.get("/models") + def list_model_types(request): + return web.json_response(list(folder_paths.folder_names_and_paths)) @routes.get("/models/{folder}") async def get_models(request):