    images = pipeline(example["prompt"]).images Next is the main() function which handles setting up the dataset for training and the training loop itself. The script loads the tokenizer, scheduler and models:   Copied # Load the tokenizer