Skip to content

Commit

Permalink
Remove errant on_prompt handler
Browse files Browse the repository at this point in the history
The orchestration code included functionality to track when execution
had completed which is not useful outside of orchestration. As the other
references to completion had been removed, this code was causing
exceptions during execution.
  • Loading branch information
AustinMroz committed Oct 17, 2024
1 parent 535ea76 commit 8a9e5d5
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 8 deletions.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
[project]
name = "comfyui-workflowcheckpointing"
description = "Automatically creates checkpoints during workflow execution. If If an workflow is canceled or ComfyUI crashes mid-execution, then these checkpoints are used when the workflow is re-queued to resume execution with minimal progress loss."
version = "1.1.0"
version = "1.1.1"
license = { file = "LICENSE" }

[project.urls]
Expand Down
7 changes: 0 additions & 7 deletions workflowcheckpointing.py
Original file line number Diff line number Diff line change
Expand Up @@ -325,13 +325,6 @@ async def fetch_remote_files(remote_files, uid=None):
if len(fetches) > 0:
await asyncio.gather(*fetches)

completion_futures = {}
def add_future(json_data):
index = max(completion_futures.keys())
json_data['extra_data']['completion_future'] = index
return json_data
server.PromptServer.instance.add_on_prompt_handler(add_future)

prompt_route = next(filter(lambda x: x.path == '/prompt' and x.method == 'POST',
server.PromptServer.instance.routes))
original_post_prompt = prompt_route.handler
Expand Down

0 comments on commit 8a9e5d5

Please sign in to comment.