) Aside from setting up the LoRA layers, the training script is more or less the same as train_text_to_image.py!  Launch the script Once you’ve made all your changes or you’re okay with the default configuration, you’re ready to launch the training script! 🚀 Let’s train on the Pokémon BLIP captions dataset to generate our yown Pokémon. Set the environment variables MODEL_NAME and DATASET_NAME to the model and dataset respectively. You should also specify where to save the model in OUTPUT_DIR, and the name of the model to save to on the Hub with HUB_MODEL_ID. The script creates and saves the following files to your repository: saved model checkpoints pytorch_lora_weights.safetensors (the trained LoRA weights) If you’re training on more than one GPU, add the --multi_gpu parameter to the accelerate launch command. A full training run takes ~5 hours on a 2080 Ti GPU with 11GB of VRAM.   Copied export MODEL_NAME="runwayml/stable-diffusion-v1-5"