tor=generator, noise_level=100).images
#stage_3_output[0].save("./if_stage_III.png")
make_image_grid([original_image, pt_to_pil(stage_1_output)[0], pt_to_pil(stage_2_output)[0], stage_3_output[0]], rows=1, rows=4)  Text Guided Inpainting Generation The same IF model weights can be used for text-guided image-to-image translation or image variation.
In this case just make sure to load the weights using the IFInpaintingPipeline and IFInpaintingSup