File size: 968 Bytes
663259c 44356d2 c1e8b26 44356d2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
---
license: mit
---
## How to run
**Note**: This is only a single prior model checkpoint and has to be run with https://huggingface.co/warp-diffusion/wuerstchen
```python
import torch
from diffusers import AutoPipelineForText2Image
from diffusers.pipelines.wuerstchen import WuerstchenPrior
prior_model = WuerstchenPrior.from_pretrained("warp-diffusion/wuerstchen-prior-model-interpolated", torch_dtype=torch.float16)
pipe = AutoPipelineForText2Image.from_pretrained("warp-diffusion/wuerstchen", prior_prior=prior_model, torch_dtype=torch.float16).to("cuda")
prompt = [
"An old destroyed car standing on a cliff in norway, cinematic photography",
"Western movie, closeup cinematic photography",
"Pink nike shoe commercial, closeup cinematic photography",
"Croatia, closeup cinematic photography",
"South Tyrol mountains at sunset, closeup cinematic photography",
]
images = pipe(prompt, guidance_scale=8.0, width=1024, height=1024).images
``` |