By now, you have most of the pieces to start training the model and all that's left is putting everything together.  
First, you'll need an optimizer and a learning rate scheduler:  
```py
>>> from diffusers.optimization import get_cosine_schedule_with_warmup