README.md

VQ Diffusion

Contribution: This model was contribution by williamberman in VQ-diffusion.

TODO: Make nice model readme

#!pip install diffusers[torch]==0.7.0 transformers
import torch
from diffusers import VQDiffusionPipeline

pipeline = VQDiffusionPipeline.from_pretrained("microsoft/vq-diffusion-ithq", torch_dtype=torch.float16, revision="fp16")
pipeline = pipeline.to("cuda")

output = pipeline("teddy bear playing in the pool", truncation_rate=0.86)

image = output.images[0]
image.save("./teddy_bear.png")