xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-12 02:26:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add list models route

Browse Source
This commit is contained in:
ImDarkTom 2024-09-08 20:18:04 +01:00
parent bb52934ba4
commit 40ddbba25c
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server.py
View File

@ -155,6 +155,10 @@ class PromptServer():
def get_embeddings(self): def get_embeddings(self):
embeddings = folder_paths.get_filename_list("embeddings") embeddings = folder_paths.get_filename_list("embeddings")
return web.json_response(list(map(lambda a: os.path.splitext(a)[0], 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}") @routes.get("/models/{folder}")
async def get_models(request): async def get_models(request):