image transforms, the script also includes transforms for the conditioning image. If you’re streaming a dataset on a TPU, performance may be bottlenecked by the 🤗 Datasets library which is not optimized for images. To ensure maximum throughput, you’re encouraged to explore other dataset formats like WebDataset, TorchData, and TensorFlow Datasets.   Copied conditioning_image_transforms = transforms.Compose(
    [
        transforms.Resize(args.r