, the edited images and and edit instructions are preprocessed and tokenized. It is important the same image transformations are applied to the original and edited images.   Copied def preprocess_train(examples):
    preprocessed_images = preprocess_images(examples)

    original_images, edited_images = preprocessed_images.chunk(2)
    original_images = original_images.reshape(-1, 3, args.resolution, args.resolution)
    edited_images = edited_