wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-08 13:20:50 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Add API endpoint to get samplers

Browse Source
This commit is contained in:
Alexis ROLLAND 2025-10-04 16:43:17 +08:00
parent bbd683098e
commit 7e781171f9
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server.py
View File

@ -26,6 +26,7 @@ import mimetypes
from comfy.cli_args import args from comfy.cli_args import args
import comfy.utils import comfy.utils
import comfy.model_management import comfy.model_management
from comfy.samplers import SAMPLER_NAMES
from comfy_api import feature_flags from comfy_api import feature_flags
import node_helpers import node_helpers
from comfyui_version import __version__ from comfyui_version import __version__
@ -286,6 +287,10 @@ class PromptServer():
return web.json_response(extensions) 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): def get_dir_by_type(dir_type):
if dir_type is None: if dir_type is None:
dir_type = "input" dir_type = "input"
@ -366,7 +371,6 @@ class PromptServer():
post = await request.post() post = await request.post()
return image_upload(post) return image_upload(post)
@routes.post("/upload/mask") @routes.post("/upload/mask")
async def upload_mask(request): async def upload_mask(request):
post = await request.post() post = await request.post()