patrickvonplaten commited on
Commit
44356d2
1 Parent(s): 59c43ad

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -0
README.md CHANGED
@@ -1,3 +1,23 @@
1
  ---
2
  license: mit
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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-model-interpolated", 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
+ ```