PixArt-alpha
/

PixArt-LCM-XL-2-1024-MS

PixArtAlphaPipeline

Model card Files Files and versions Community

Lawrence-cj commited on Dec 7, 2023

Commit

e153060

•

1 Parent(s): 534bd5f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -90,7 +90,7 @@ pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-LCM-XL-2-1024-MS
 pipe.enable_model_cpu_offload()
 prompt = "A small cactus with a happy face in the Sahara desert."
-image = pipe(prompt).images[0]
 ```
 When using `torch >= 2.0`, you can improve the inference speed by 20-30% with torch.compile. Simple wrap the unet with torch compile before running the pipeline:

 pipe.enable_model_cpu_offload()
 prompt = "A small cactus with a happy face in the Sahara desert."
+image = pipe(prompt, guidance_scale=0., num_inference_steps=4).images[0]
 ```
 When using `torch >= 2.0`, you can improve the inference speed by 20-30% with torch.compile. Simple wrap the unet with torch compile before running the pipeline: