ion model) is conditioned on this highly compressed space to generate an image. To fit the prior model into GPU memory and to speedup training, try enabling gradient_accumulation_steps, gradient_checkpointing, and mixed_precision respectively. This guide explores the train_text_to_image_prior.py script to help you become more familiar with it, and how you can adapt it for your own use-case. Before running the script, make sure you install the l