diff --git a/server.py b/server.py index c27f8be7d..209d4b53e 100644 --- a/server.py +++ b/server.py @@ -28,6 +28,7 @@ import mimetypes from comfy.cli_args import args import comfy.utils import comfy.model_management +from comfy.samplers import SAMPLER_NAMES from comfy_api import feature_flags import node_helpers from comfyui_version import __version__ @@ -343,6 +344,10 @@ class PromptServer(): return web.json_response(extensions) + @routes.get("/samplers") + def get_samplers(request): + return web.json_response(sorted(SAMPLER_NAMES)) + def get_dir_by_type(dir_type): if dir_type is None: dir_type = "input" @@ -423,7 +428,6 @@ class PromptServer(): post = await request.post() return image_upload(post) - @routes.post("/upload/mask") async def upload_mask(request): post = await request.post()