xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-13 12:27:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert 'Move model manager routes'

Browse Source
This commit is contained in:
hayden 2024-12-03 09:26:04 +08:00
parent 63a272d172
commit c2da0dadca
2 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
model_filemanager/model_filemanager.py
View File

@ -23,20 +23,6 @@ class ModelFileManager:
self.cache.clear() self.cache.clear()
def add_routes(self, routes): def add_routes(self, routes):
@routes.get("/models")
def list_model_types(request):
model_types = list(folder_paths.folder_names_and_paths.keys())
return web.json_response(model_types)
@routes.get("/models/{folder}")
async def get_models(request):
folder = request.match_info.get("folder", None)
if not folder in folder_paths.folder_names_and_paths:
return web.Response(status=404)
files = folder_paths.get_filename_list(folder)
return web.json_response(files)
# NOTE: This is an experiment to replace `/models` # NOTE: This is an experiment to replace `/models`
@routes.get("/experiment/models") @routes.get("/experiment/models")
async def get_model_folders(request): async def get_model_folders(request):

14
server.py
View File

@ -223,6 +223,20 @@ class PromptServer():
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):
model_types = list(folder_paths.folder_names_and_paths.keys())
return web.json_response(model_types)
@routes.get("/models/{folder}")
async def get_models(request):
folder = request.match_info.get("folder", None)
if not folder in folder_paths.folder_names_and_paths:
return web.Response(status=404)
files = folder_paths.get_filename_list(folder)
return web.json_response(files)
@routes.get("/extensions") @routes.get("/extensions")
async def get_extensions(request): async def get_extensions(request):
files = glob.glob(os.path.join( files = glob.glob(os.path.join(