Luo-Yihong
/

yoso_pixart1024

Model card Files Files and versions Community

Yihong Luo commited on Mar 18

Commit

a4063f4

•

1 Parent(s): ff9d7df

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -9,7 +9,7 @@ This model is fine-tuning from PixArt, enabling one-step inference to perform te
 from diffusers import PixArtAlphaPipeline, LCMScheduler, Transformer2DModel, DPMSolverMultistepScheduler
 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,

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