ded in the Text-to-image training guide. Instead, this guide takes a look at the relevant parts of the ControlNet script. The training script has a make_train_dataset function for preprocessing the dataset with image transforms and caption tokenization. You’ll see that in addition to the usual caption tokenization and image transforms, the script also includes transforms for the conditioning image. If you’re streaming a dataset on a TPU, perfor