Set the `MODEL_NAME` environment variable to the name of the model (can be a model id on the Hub or a path to a local model), and the `DATASET_ID` to the name of the dataset on the Hub. The script creates and saves all the components (feature extractor, scheduler, text encoder, UNet, etc.) to a subfolder in your repository.  
<Tip>  
For better results, try longer training runs with a larger dataset. We've only tested this training script on a smaller-scale dataset.  
<br>