Update README.md
Browse files
README.md
CHANGED
|
@@ -35,7 +35,7 @@ This pipeline should be run together with https://huggingface.co/warp-ai/wuerstc
|
|
| 35 |
```py
|
| 36 |
import torch
|
| 37 |
from diffusers import WuerstchenDecoderPipeline, WuerstchenPriorPipeline
|
| 38 |
-
from diffusers.pipelines.wuerstchen import WuerstchenPrior,
|
| 39 |
|
| 40 |
device = "cuda"
|
| 41 |
dtype = torch.float16
|
|
@@ -56,7 +56,7 @@ prior_output = prior_pipeline(
|
|
| 56 |
prompt=caption,
|
| 57 |
height=1024,
|
| 58 |
width=1024,
|
| 59 |
-
timesteps=
|
| 60 |
negative_prompt=negative_prompt,
|
| 61 |
guidance_scale=4.0,
|
| 62 |
num_images_per_prompt=num_images_per_prompt,
|
|
@@ -65,7 +65,6 @@ decoder_output = decoder_pipeline(
|
|
| 65 |
image_embeddings=prior_output.image_embeddings,
|
| 66 |
prompt=caption,
|
| 67 |
negative_prompt=negative_prompt,
|
| 68 |
-
num_images_per_prompt=num_images_per_prompt,
|
| 69 |
guidance_scale=0.0,
|
| 70 |
output_type="pil",
|
| 71 |
).images
|
|
|
|
| 35 |
```py
|
| 36 |
import torch
|
| 37 |
from diffusers import WuerstchenDecoderPipeline, WuerstchenPriorPipeline
|
| 38 |
+
from diffusers.pipelines.wuerstchen import WuerstchenPrior, DEFAULT_STAGE_C_TIMESTEPS
|
| 39 |
|
| 40 |
device = "cuda"
|
| 41 |
dtype = torch.float16
|
|
|
|
| 56 |
prompt=caption,
|
| 57 |
height=1024,
|
| 58 |
width=1024,
|
| 59 |
+
timesteps=DEFAULT_STAGE_C_TIMESTEPS,
|
| 60 |
negative_prompt=negative_prompt,
|
| 61 |
guidance_scale=4.0,
|
| 62 |
num_images_per_prompt=num_images_per_prompt,
|
|
|
|
| 65 |
image_embeddings=prior_output.image_embeddings,
|
| 66 |
prompt=caption,
|
| 67 |
negative_prompt=negative_prompt,
|
|
|
|
| 68 |
guidance_scale=0.0,
|
| 69 |
output_type="pil",
|
| 70 |
).images
|