[nodes] Fixes calls into image to image and inpaint from nodes (#2940)

This commit is contained in:
blessedcoolant 2023-03-13 19:05:32 +13:00 committed by GitHub
commit 7fe2606cb3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -118,7 +118,7 @@ class ImageToImageInvocation(TextToImageInvocation):
generator_output = next(
Img2Img(model).generate(
prompt=self.prompt,
init_img=image,
init_image=image,
init_mask=mask,
step_callback=step_callback,
**self.dict(
@ -179,8 +179,8 @@ class InpaintInvocation(ImageToImageInvocation):
generator_output = next(
Inpaint(model).generate(
prompt=self.prompt,
init_img=image,
init_mask=mask,
init_image=image,
mask_image=mask,
step_callback=step_callback,
**self.dict(
exclude={"prompt", "image", "mask"}