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

feat(nodes): provide board_id in image creation

Browse Source
This commit is contained in:
psychedelicious 2023-09-22 21:59:29 +10:00 committed by Kent Keirsey
parent 88c8cb61f0
commit 60b3c6a201
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/app/invocations/image.py
View File

@ -993,6 +993,7 @@ class SaveImageInvocation(BaseInvocation):
image=image, image=image,
image_origin=ResourceOrigin.INTERNAL, image_origin=ResourceOrigin.INTERNAL,
image_category=ImageCategory.GENERAL, image_category=ImageCategory.GENERAL,
board_id=self.board.board_id if self.board else None,
node_id=self.id, node_id=self.id,
session_id=context.graph_execution_state_id, session_id=context.graph_execution_state_id,
is_intermediate=self.is_intermediate, is_intermediate=self.is_intermediate,
@ -1000,12 +1001,6 @@ class SaveImageInvocation(BaseInvocation):
workflow=self.workflow, workflow=self.workflow,
) )
if self.board:
context.services.board_images.add_image_to_board(
board_id=self.board.board_id,
image_name=image_dto.image_name,
)
return ImageOutput( return ImageOutput(
image=ImageField(image_name=image_dto.image_name), image=ImageField(image_name=image_dto.image_name),
width=image_dto.width, width=image_dto.width,