Diffusers
Safetensors
WuerstchenPriorPipeline
patrickvonplaten commited on
Commit
65b9bdd
1 Parent(s): 9def71b

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -0
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
+ ```