From 95ecb1a0c1033820cd3d221f4cd4f503ce3b5c9f Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Tue, 5 Sep 2023 12:30:00 +1000 Subject: [PATCH] fix(ip_adapter): add None to types --- invokeai/backend/ip_adapter/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/invokeai/backend/ip_adapter/utils.py b/invokeai/backend/ip_adapter/utils.py index d43cf12154..b500fd2570 100644 --- a/invokeai/backend/ip_adapter/utils.py +++ b/invokeai/backend/ip_adapter/utils.py @@ -20,7 +20,7 @@ def is_torch2_available(): @torch.no_grad() def generate( self, - prompt: Union[str, List[str]] = None, + prompt: Union[str, List[str], None] = None, image: Union[ torch.FloatTensor, PIL.Image.Image, @@ -28,6 +28,7 @@ def generate( List[torch.FloatTensor], List[PIL.Image.Image], List[np.ndarray], + None, ] = None, height: Optional[int] = None, width: Optional[int] = None,