[nodes] Fixes calls into image to image and inpaint from nodes

This commit is contained in:
Kyle Schouviller 2023-03-12 22:12:42 -07:00
parent 60614badaf
commit 0c3b1fe3c4

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"}