U with at least 30GB of memory.
Before running the script, make sure you have the requirements installed:


	Copied
pip install -U -r requirements.txt
Specify the MODEL_NAME environment variable (either a Hub model repository id or a path to the directory containing the model weights) and pass it to the ~diffusers.DiffusionPipeline.from_pretrained.pretrained_model_name_or_path argument.
Now you can launch the training script with the following 