 For smaller datasets, the effects of Min-SNR may not be as obvious compared to larger datasets.  Training script The dataset preprocessing code and training loop are found in the main() function. If you need to adapt the training script, this is where you’ll need to make your changes. The train_text_to_image script starts by loading a scheduler and tokenizer. You can choose to use a different scheduler here if you want:   Copied noise_schedule