Simplify Wuerstchen model card
#1
by
patrickvonplaten
- opened
README.md
CHANGED
@@ -35,39 +35,25 @@ This pipeline should be run together with a prior https://huggingface.co/warp-ai
|
|
35 |
|
36 |
```py
|
37 |
import torch
|
38 |
-
from diffusers import
|
39 |
-
from diffusers.pipelines.wuerstchen import default_stage_c_timesteps
|
40 |
|
41 |
device = "cuda"
|
42 |
dtype = torch.float16
|
43 |
-
num_images_per_prompt = 2
|
44 |
|
45 |
-
|
46 |
-
"warp-
|
47 |
-
).to(device)
|
48 |
-
decoder_pipeline = WuerstchenDecoderPipeline.from_pretrained(
|
49 |
-
"warp-ai/wuerstchen", torch_dtype=dtype
|
50 |
).to(device)
|
51 |
|
52 |
caption = "Anthropomorphic cat dressed as a fire fighter"
|
53 |
negative_prompt = ""
|
54 |
|
55 |
-
|
56 |
prompt=caption,
|
57 |
height=1024,
|
58 |
-
width=
|
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,
|
63 |
-
)
|
64 |
-
decoder_output = decoder_pipeline(
|
65 |
-
image_embeddings=prior_output.image_embeddings,
|
66 |
-
prompt=caption,
|
67 |
negative_prompt=negative_prompt,
|
68 |
-
|
69 |
-
|
70 |
-
output_type="pil",
|
71 |
).images
|
72 |
```
|
73 |
|
|
|
35 |
|
36 |
```py
|
37 |
import torch
|
38 |
+
from diffusers import AutoPipelineForText2Image
|
|
|
39 |
|
40 |
device = "cuda"
|
41 |
dtype = torch.float16
|
|
|
42 |
|
43 |
+
pipeline = AutoPipelineForText2Image.from_pretrained(
|
44 |
+
"warp-diffusion/wuerstchen", torch_dtype=dtype
|
|
|
|
|
|
|
45 |
).to(device)
|
46 |
|
47 |
caption = "Anthropomorphic cat dressed as a fire fighter"
|
48 |
negative_prompt = ""
|
49 |
|
50 |
+
output = pipeline(
|
51 |
prompt=caption,
|
52 |
height=1024,
|
53 |
+
width=1024,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
54 |
negative_prompt=negative_prompt,
|
55 |
+
prior_guidance_scale=4.0,
|
56 |
+
decoder_guidance_scale=0.0,
|
|
|
57 |
).images
|
58 |
```
|
59 |
|