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