microsoft
/

vq-diffusion-ithq

VQDiffusionPipeline

Model card Files Files and versions

patrickvonplaten commited on Nov 16, 2022

Commit

56f5cfa

·

1 Parent(s): 166ec28

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -15,7 +15,7 @@ license: mit
 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)

 import torch
 from diffusers import VQDiffusionPipeline
+pipeline = VQDiffusionPipeline.from_pretrained("microsoft/vq-diffusion-ithq", torch_dtype=torch.float16)
 pipeline = pipeline.to("cuda")
 output = pipeline("teddy bear playing in the pool", truncation_rate=0.86)