make_image_grid([raw_image, mask_image, output_image], rows=1, cols=3)  original image  edited image  Generate source and target embeddings The source and target embeddings can be automatically generated with the Flan-T5 model instead of creating them manually. Load the Flan-T5 model and tokenizer from the 🤗 Transformers library:   Copied import torch