Update README.md

README.md

@@ -2,12 +2,14 @@
 
 This algorithm was proposed in You Only Sample Once: Taming One-Step Text-To-Image Synthesis by Self-Cooperative Diffusion GANs. 
 
-This model is fine-tuning from PixArt, enabling one-step inference to perform text-to-image generation.
+This model is fine-tuning from [
+PixArt-XL-2-512x512](https://huggingface.co/PixArt-alpha/PixArt-XL-2-512x512), enabling one-step inference to perform text-to-image generation.
 
-We wanna highlight that the YOSO-PixArt was originally trained on 512 resolution. However we found that we can construct a YOSO that enables generating samples with 1024 resolution by merging with PixArt-1024 (Eq(15) in the paper) as follows:
+We wanna highlight that the YOSO-PixArt was originally trained on 512 resolution. However, we found that we can construct a YOSO that enables generating samples with 1024 resolution by merging with [
+PixArt-XL-2-1024-MS](https://huggingface.co/PixArt-alpha/PixArt-XL-2-1024-MS
+) (Eq(15) in the paper) as follows:
 ![Construction](construction.jpg)
 The impressive performance indicates the robust generalization ability of our YOSO. 
-
 ## usage
 ```python
 import torch