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