patrickvonplaten
commited on
Commit
•
65b9bdd
1
Parent(s):
9def71b
Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: mit
|
3 |
+
---
|
4 |
+
|
5 |
+
## How to run
|
6 |
+
|
7 |
+
```python
|
8 |
+
import torch
|
9 |
+
from diffusers import AutoPipelineForText2Image
|
10 |
+
from diffusers.pipelines.wuerstchen import WuerstchenPrior
|
11 |
+
|
12 |
+
prior_model = WuerstchenPrior.from_pretrained("warp-diffusion/wuerstchen-prior", torch_dtype=torch.float16)
|
13 |
+
pipe = AutoPipelineForText2Image.from_pretrained("warp-diffusion/wuerstchen", prior_prior=prior_model, torch_dtype=torch.float16).to("cuda")
|
14 |
+
|
15 |
+
prompt = [
|
16 |
+
"An old destroyed car standing on a cliff in norway, cinematic photography",
|
17 |
+
"Western movie, closeup cinematic photography",
|
18 |
+
"Pink nike shoe commercial, closeup cinematic photography",
|
19 |
+
"Croatia, closeup cinematic photography",
|
20 |
+
"South Tyrol mountains at sunset, closeup cinematic photography",
|
21 |
+
]
|
22 |
+
images = pipe(prompt, guidance_scale=8.0, width=1024, height=1024).images
|
23 |
+
```
|