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

Fix distributed prompting

Browse Source
This commit is contained in:
doctorpangloss 2024-02-16 06:17:06 -08:00
parent 0546b01080
commit d29028dd3e
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/cmd/server.py
View File

@ -22,7 +22,7 @@ import aiofiles
import aiohttp import aiohttp
from aiohttp import web from aiohttp import web
from ..component_model.queue_types import QueueItem, HistoryEntry, BinaryEventTypes from ..component_model.queue_types import QueueItem, HistoryEntry, BinaryEventTypes, TaskInvocation
from ..cmd import execution from ..cmd import execution
from ..cmd import folder_paths from ..cmd import folder_paths
import mimetypes import mimetypes
@ -621,7 +621,7 @@ class PromptServer(ExecutorToClientProgress):
# todo: check that the files specified in the InputFile nodes exist # todo: check that the files specified in the InputFile nodes exist
# convert a valid prompt to the queue tuple this expects # convert a valid prompt to the queue tuple this expects
completed: Future = self.loop.create_future() completed: Future[TaskInvocation | dict] = self.loop.create_future()
number = self.number number = self.number
self.number += 1 self.number += 1
self.prompt_queue.put( self.prompt_queue.put(
@ -633,7 +633,8 @@ class PromptServer(ExecutorToClientProgress):
except Exception as ex: except Exception as ex:
return web.Response(body=str(ex), status=503) return web.Response(body=str(ex), status=503)
# expect a single image # expect a single image
outputs_dict: dict = completed.result() result: TaskInvocation | dict = completed.result()
outputs_dict: dict = result.outputs if isinstance(result, TaskInvocation) else result
# find images and read them # find images and read them
output_images: List[str] = [] output_images: List[str] = []