From 87f651ea6758b6939fb8cf4e2fc5478efe94a15d Mon Sep 17 00:00:00 2001 From: huchenlei Date: Thu, 22 Aug 2024 16:55:52 -0400 Subject: [PATCH] Fix task.status.status_str caused by 2666 regression --- execution.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/execution.py b/execution.py index 05a662cde..99f2757af 100644 --- a/execution.py +++ b/execution.py @@ -450,7 +450,7 @@ class PromptExecutor: "current_outputs": list(current_outputs), } self.add_message("execution_error", mes, broadcast=False) - + def execute(self, prompt, prompt_id, extra_data={}, execute_outputs=[]): nodes.interrupt_processing(False) @@ -494,6 +494,7 @@ class PromptExecutor: result, error, ex = execute(self.server, dynamic_prompt, self.caches, node_id, extra_data, executed, prompt_id, execution_list, pending_subgraph_results) if result == ExecutionResult.FAILURE: self.handle_execution_error(prompt_id, dynamic_prompt.original_prompt, current_outputs, executed, error, ex) + self.success = False break elif result == ExecutionResult.PENDING: execution_list.unstage_node_execution()