Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(nodes): fix get_workflow from queue item dict func

Browse Source
This commit is contained in:
psychedelicious 2023-11-29 15:36:28 +11:00
parent 14bf87e5e7
commit c1bfc1f47b
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/app/services/session_queue/session_queue_common.py
View File

@ -171,10 +171,12 @@ def get_session(queue_item_dict: dict) -> GraphExecutionState:
return session return session
def get_workflow(queue_item_dict: dict) -> WorkflowWithoutID: def get_workflow(queue_item_dict: dict) -> Optional[WorkflowWithoutID]:
workflow_raw = queue_item_dict.get("workflow", "{}") workflow_raw = queue_item_dict.get("workflow", None)
if workflow_raw is not None:
workflow = WorkflowWithoutIDValidator.validate_json(workflow_raw, strict=False) workflow = WorkflowWithoutIDValidator.validate_json(workflow_raw, strict=False)
return workflow return workflow
return None
class SessionQueueItemWithoutGraph(BaseModel): class SessionQueueItemWithoutGraph(BaseModel):