wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-12 22:42:36 +08:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

add get model list to api

Browse Source
This commit is contained in:
flipnism 2023-09-01 08:17:45 +07:00
parent cfe1c54de8
commit 2ba4e2ffbb
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
server.py
View File

@ -124,6 +124,15 @@ class PromptServer():
async def get_root(request): async def get_root(request):
return web.FileResponse(os.path.join(self.web_root, "index.html")) return web.FileResponse(os.path.join(self.web_root, "index.html"))
@routes.get("/models")
def get_models_lists(request):
which_models = request.rel_url.query.get("model", "")
try:
models_dir = folder_paths.get_filename_list(which_models.lower())
return web.json_response(list(models_dir))
except:
return web.json_response(list([""]))
@routes.get("/embeddings") @routes.get("/embeddings")
def get_embeddings(self): def get_embeddings(self):
embeddings = folder_paths.get_filename_list("embeddings") embeddings = folder_paths.get_filename_list("embeddings")