From c1bfc1f47b0f1958d762600a488630ebe4f831c2 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Wed, 29 Nov 2023 15:36:28 +1100 Subject: [PATCH] fix(nodes): fix get_workflow from queue item dict func --- .../app/services/session_queue/session_queue_common.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/invokeai/app/services/session_queue/session_queue_common.py b/invokeai/app/services/session_queue/session_queue_common.py index e8e06f9194..94db6999c2 100644 --- a/invokeai/app/services/session_queue/session_queue_common.py +++ b/invokeai/app/services/session_queue/session_queue_common.py @@ -171,10 +171,12 @@ def get_session(queue_item_dict: dict) -> GraphExecutionState: return session -def get_workflow(queue_item_dict: dict) -> WorkflowWithoutID: - workflow_raw = queue_item_dict.get("workflow", "{}") - workflow = WorkflowWithoutIDValidator.validate_json(workflow_raw, strict=False) - return workflow +def get_workflow(queue_item_dict: dict) -> Optional[WorkflowWithoutID]: + workflow_raw = queue_item_dict.get("workflow", None) + if workflow_raw is not None: + workflow = WorkflowWithoutIDValidator.validate_json(workflow_raw, strict=False) + return workflow + return None class SessionQueueItemWithoutGraph(BaseModel):