Luo-Yihong
/

yoso_pixart1024

Model card Files Files and versions Community

Yihong Luo commited on Mar 18

Commit

a662069

•

1 Parent(s): a4063f4

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -11,7 +11,7 @@ from diffusers import PixArtAlphaPipeline, LCMScheduler, Transformer2DModel, DPM
 transformer = Transformer2DModel.from_pretrained(
     "Yihong666/yoso_pixart1024", torch_dtype=torch.float16).to('cuda')
-pipe = PixArtAlphaPipeline.from_pretrained("./models--PixArt-alpha--PixArt-XL-2-512x512",
                                            transformer=transformer,
                                            torch_dtype=torch.float16, use_safetensors=True)

 transformer = Transformer2DModel.from_pretrained(
     "Yihong666/yoso_pixart1024", torch_dtype=torch.float16).to('cuda')
+pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-512x512",
                                            transformer=transformer,
                                            torch_dtype=torch.float16, use_safetensors=True)