make_image_grid([pt_to_pil(stage_1_output)[0], pt_to_pil(stage_2_output)[0], stage_3_output[0]], rows=1, rows=3)  Text Guided Image-to-Image Generation The same IF model weights can be used for text-guided image-to-image translation or image variation.