update README.md

README.md

@@ -11,12 +11,12 @@ tags:
 - control-lora
 inference: true
 ---
-# ControlLoRA text2image fine-tuning - Official Repository
-These are ControlLoRA adaption weights for [runwayml/stable-diffusion-v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5). 
+# ControlLoRA text2image fine-tuning - Official Model Repository
+These are ControlLoRA adaption weights for [runwayml/stable-diffusion-v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5). You can find the code repository in [HighCWu/ControlLoRA](https://github.com/HighCWu/ControlLoRA).
 
 With ControlLoRA, a simple and small (~7M parameters, ~25M storage space) network, you could control the spatial information of stable diffusion.
 
-Each of the weights is fine-tuned on the [fill50k](https://huggingface.co/datasets/HighCWu/fill50k), [diffusiondb_2m_first_5k_canny](https://huggingface.co/datasets/HighCWu/diffusiondb_2m_first_5k_canny) and [mpii_100_openpose](https://huggingface.co/datasets/HighCWu/mpii_100_openpose) datasets with 30k steps on RTX4080 in 3 hours. 
+Each of the weights is fine-tuned on the [diffusiondb_2m_first_5k_canny](https://huggingface.co/datasets/HighCWu/diffusiondb_2m_first_5k_canny) and [mpii_100_openpose](https://huggingface.co/datasets/HighCWu/mpii_100_openpose) datasets with 30k steps on RTX4080 in 3 hours. 
 
 You can find some example images in the following. 
 
@@ -36,3 +36,6 @@ beautiful - n 9, by Eric Dinyer, young child, midlands*
 "vladimir kush, giger, roger dean, 8 k*
 
 I also uploaded a lora model fine-tuned on my selfies with 2k steps which could be use in the experiment of mixing LoRA and ControlLoRA.
+
+![img_{i}](./docs/imgs/7.png)
+*portrait of female HighCWu as a cute pink hair girl*