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

Update readme, remove unused import

Browse Source
This commit is contained in:
doctorpangloss 2024-02-08 10:09:47 -08:00
parent 5ace1e9fee
commit 09838ed604
2 changed files with 24 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/app/user_manager.py
View File

@ -2,7 +2,6 @@ import json
import os import os
import re import re
import uuid import uuid
from typing import Optional
from aiohttp import web from aiohttp import web
from ..cli_args import args from ..cli_args import args

28
comfy/client/embedded_comfy_client.py
View File

@ -15,6 +15,9 @@ from ..component_model.executor_types import ExecutorToClientProgress, StatusMes
class ServerStub(ExecutorToClientProgress): class ServerStub(ExecutorToClientProgress):
"""
This class is a stub implementation of ExecutorToClientProgress. This will handle progress events.
"""
def __init__(self): def __init__(self):
self.client_id = str(uuid.uuid4()) self.client_id = str(uuid.uuid4())
self.last_node_id = None self.last_node_id = None
@ -41,12 +44,29 @@ class EmbeddedComfyClient:
Asynchronous (non-blocking) usage with async-await: Asynchronous (non-blocking) usage with async-await:
``` ```
prompt = dict() # ... # Write a workflow, or enable Dev Mode in the UI settings, then Save (API Format) to get the workflow in your
async with EmbeddedComfyClient() as client: # workspace.
outputs = await client.queue_prompt(prompt) prompt_dict = {
"1": {"class_type": "KSamplerAdvanced", ...}
...
}
print(outputs) # Validate your workflow (the prompt)
from comfy.api.components.schema.prompt import Prompt
prompt = Prompt.validate(prompt_dict)
# Then use the client to run your workflow. This will start, then stop, a local ComfyUI workflow executor.
# It does not connect to a remote server.
async def main():
async with EmbeddedComfyClient() as client:
outputs = await client.queue_prompt(prompt)
print(outputs)
print("Now that we've exited the with statement, all your VRAM has been cleared from ComfyUI")
if __name__ == "__main__"
asyncio.run(main())
``` ```
In order to use this in blocking methods, learn more about asyncio online.
""" """
def __init__(self, configuration: Optional[Configuration] = None, loop: Optional[AbstractEventLoop] = None, def __init__(self, configuration: Optional[Configuration] = None, loop: Optional[AbstractEventLoop] = None,