 works, check out the Understanding pipelines, models and schedulers tutorial which breaks down the basic pattern of the denoising process.  Launch the script Now you’re ready to launch the training script! 🚀 For this example training, you’ll use the fusing/fill50k dataset. You can also create and use your own dataset if you want (see the Create a dataset for training guide). Set the environment variable MODEL_DIR to a model id on the Hub or a 