Update README.md

README.md

@@ -25,7 +25,12 @@ However, aurora is so rare that we can hardly see it even in the very north plac
 Don't worry. Now we have generative models!!!
 
 # Model Details
-This model is trained from 
+
+This model is trained from dataset [huggan/few-shot-aurora](https://huggingface.co/datasets/huggan/few-shot-aurora).
+
+The training method is modified from this [example](https://colab.sandbox.google.com/github/huggingface/notebooks/blob/main/diffusers/training_example.ipynb).
+
+You can check my training source code here: [<img src="https://colab.research.google.com/assets/colab-badge.svg">](https://colab.sandbox.google.com/github/Li-Yan/Diffusion-Model/blob/main/li_yan_diffusers_training_accelerate.ipynb)
 
 # Usage