wuerstchen / README.md
patrickvonplaten's picture
Update README.md
5d1ddad
|
raw
history blame
892 Bytes
metadata
license: apache-2.0
prior:
  - warp-diffusion/wuerstchen-prior
tags:
  - text-to-image
  - wuerstchen

How-to-use

By using this pipeline with AutoPipelineForText2Image, the required prior pipeline: https://huggingface.co/warp-diffusion/wuerstchen-prior will automatically be downloaded and run.

import torch
from diffusers import AutoPipelineForText2Image

pipe = AutoPipelineForText2Image.from_pretrained("warp-diffusion/wuerstchen", 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