From ae3d7218c98407d4fc3bc9d3ac902fbadf34d241 Mon Sep 17 00:00:00 2001 From: maryhipp Date: Tue, 16 Jan 2024 12:06:36 -0500 Subject: [PATCH] forward optional kwargs through image service create --- invokeai/app/services/images/images_base.py | 1 + invokeai/app/services/images/images_default.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/invokeai/app/services/images/images_base.py b/invokeai/app/services/images/images_base.py index b3990d08f5..88eacdfa33 100644 --- a/invokeai/app/services/images/images_base.py +++ b/invokeai/app/services/images/images_base.py @@ -52,6 +52,7 @@ class ImageServiceABC(ABC): is_intermediate: Optional[bool] = False, metadata: Optional[MetadataField] = None, workflow: Optional[WorkflowField] = None, + **kwargs ) -> ImageDTO: """Creates an image, storing the file and its metadata.""" pass diff --git a/invokeai/app/services/images/images_default.py b/invokeai/app/services/images/images_default.py index 63fa78d6c8..e592ba5c4d 100644 --- a/invokeai/app/services/images/images_default.py +++ b/invokeai/app/services/images/images_default.py @@ -43,6 +43,7 @@ class ImageService(ImageServiceABC): is_intermediate: Optional[bool] = False, metadata: Optional[MetadataField] = None, workflow: Optional[WorkflowField] = None, + **kwargs ) -> ImageDTO: if image_origin not in ResourceOrigin: raise InvalidOriginException @@ -75,6 +76,7 @@ class ImageService(ImageServiceABC): node_id=node_id, metadata=metadata, session_id=session_id, + **kwargs ) if board_id is not None: self.__invoker.services.board_image_records.add_image_to_board(board_id=board_id, image_name=image_name)