Commit
•
e153060
1
Parent(s):
534bd5f
Update README.md
Browse files
README.md
CHANGED
|
@@ -90,7 +90,7 @@ pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-LCM-XL-2-1024-MS
|
|
| 90 |
pipe.enable_model_cpu_offload()
|
| 91 |
|
| 92 |
prompt = "A small cactus with a happy face in the Sahara desert."
|
| 93 |
-
image = pipe(prompt).images[0]
|
| 94 |
```
|
| 95 |
|
| 96 |
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:
|
|
|
|
| 90 |
pipe.enable_model_cpu_offload()
|
| 91 |
|
| 92 |
prompt = "A small cactus with a happy face in the Sahara desert."
|
| 93 |
+
image = pipe(prompt, guidance_scale=0., num_inference_steps=4).images[0]
|
| 94 |
```
|
| 95 |
|
| 96 |
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:
|