-image training guide, but it’s been modified to support training the prior and decoder models. This guide focuses on the code that is unique to the Kandinsky 2.2 training scripts. prior model decoder model  The main() function contains the code for preparing the dataset and training the model. One of the main differences you’ll notice right away is that the training script also loads a CLIPImageProcessor - in addition to a scheduler and tokeni