Luo-Yihong
/

yoso_pixart1024

Model card Files Files and versions Community

Luo-Yihong commited on Mar 19, 2024

Commit

27feebe

•

1 Parent(s): b9aa955

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ import torch
 from diffusers import PixArtAlphaPipeline, LCMScheduler, Transformer2DModel, DPMSolverMultistepScheduler
 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,

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