to the ~diffusers.DiffusionPipeline.from_pretrained.pretrained_model_name_or_path argument, and the DATA_DIR environment variable to the path of the directory containing the images. 
Now you can launch the training script:
💡 A full training run takes ~1 hour on one V100 GPU. While you’re waiting for the training to complete, feel free to check out how Textual Inversion works in the section below if you’re curious!


Pytorch

Hide Pytorch conten