ompt_embeds=negative_embeds,
    output_type="pt",
    generator=generator,
).images

#pt_to_pil(stage_2_output)[0].save("./if_stage_II.png")
make_image_grid([pt_to_pil(stage_1_output)[0], pt_to_pil(stage_2_output)[0]], rows=1, rows=2)  Available Pipelines: Pipeline Tasks Colab pipeline_if.py Text-to-Image Generation - pipeline_if_superresolution.py Text-to-Image Generation - pipeline_if_img2img.py Image-to-Image Generation - pipeline_if_img2im