github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04: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
parent faabfe9520
commit f59c3f560a
1 changed files with 1 additions and 6 deletions
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_origin=ResourceOrigin.INTERNAL,
image_category=ImageCategory.GENERAL,
board_id=self.board.board_id if self.board else None,
node_id=self.id,
session_id=context.graph_execution_state_id,
is_intermediate=self.is_intermediate,
@@ -1000,12 +1001,6 @@ class SaveImageInvocation(BaseInvocation):
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(
image=ImageField(image_name=image_dto.image_name),
width=image_dto.width,