Update README.md

README.md

@@ -19,6 +19,13 @@ These are comparion for this model ran on T4 GPU, compared with segmind/tiny-sd,
 
 ![image](./compare.png)
 
+# Training info
+
+Learning rate: 5e-5 (0.00005) <br>
+Batch size: 8 <br>
+Number of steps: 1000 <br>
+Training methond: Knowledge Distillation <br>
+Trained on: x1 T4 GPU <br>
 
 ### THIS IS NOT THE FULL DESCRIPTION FOR THIS MODEL.
 
@@ -32,7 +39,7 @@ The CreativeML OpenRAIL License specifies:
 3. You may re-distribute the weights and use the model commercially and/or as a service. If you do, please be aware you have to include the same use restrictions as the ones in the license and share a copy of the CreativeML OpenRAIL-M to all your users (please read the license entirely and carefully)
 [Please read the full license here](https://huggingface.co/spaces/CompVis/stable-diffusion-license)
 
-Code:
+# Example coe:
 ```python
 import torch
 from diffusers import StableDiffusionPipeline