Luo-Yihong
/

yoso_pixart512

Model card Files Files and versions Community

Yihong Luo commited on Mar 18, 2024

Commit

5a544b4

·

verified ·

1 Parent(s): f1113ce

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_pixart512", 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_pixart512", 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)