298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge c3cc84fa18 into 2307ff6746

Browse Source
This commit is contained in:
Dr.Lt.Data 2025-01-08 19:17:00 -05:00 committed by GitHub
commit f23554e8a8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
server.py
View File

@ -33,6 +33,7 @@ from app.model_manager import ModelFileManager
from app.custom_node_manager import CustomNodeManager
from typing import Optional
from api_server.routes.internal.internal_routes import InternalRoutes
import latent_preview
class BinaryEventTypes:
PREVIEW_IMAGE = 1
@ -692,6 +693,26 @@ class PromptServer():
return web.Response(status=200)
@routes.post("/preview_method")
async def preview_method(request):
json_data = await request.json()
if "value" in json_data:
method = json_data['value']
if method == 'auto':
args.preview_method = latent_preview.LatentPreviewMethod.Auto
elif method == 'latent2rgb':
args.preview_method = latent_preview.LatentPreviewMethod.Latent2RGB
elif method == 'taesd':
args.preview_method = latent_preview.LatentPreviewMethod.TAESD
else:
args.preview_method = latent_preview.LatentPreviewMethod.NoPreviews
return web.Response(status=200)
return web.Response(status=400)
async def setup(self):
timeout = aiohttp.ClientTimeout(total=None) # no timeout
self.client_session = aiohttp.ClientSession(timeout=timeout)