ion-inpainting",
    torch_dtype=torch.float16,
)
pipe = pipe.to("cuda")

prompt = "Face of a yellow cat, high resolution, sitting on a park bench"
image = pipe(prompt=prompt, image=init_image, mask_image=mask_image).images[0]
image
mask_image
prompt
Output


Face of a yellow cat, high resolution, sitting on a park bench

You can also run this example on colab 
A previous experimental implementation of in-painting used a different, lower-qualit