Update README.md

README.md

@@ -19,7 +19,7 @@ import torch
 from diffusers import PixArtAlphaPipeline, LCMScheduler, Transformer2DModel
 
 transformer = Transformer2DModel.from_pretrained(
-    "Luo-Yihong/yoso_pixart1024", torch_dtype=torch.float16).to('cuda')
+    "Luo-Yihong/yoso_pixart512", torch_dtype=torch.float16).to('cuda')
 
 pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-512x512", 
                                            transformer=transformer,