From 3fec3b580454491dab6c692f5d160f2ea58f9755 Mon Sep 17 00:00:00 2001 From: Peter Lorenz Date: Thu, 1 Jun 2023 20:05:25 +0200 Subject: [PATCH] Update readme.md Following this issue: https://github.com/microsoft/VQ-Diffusion/issues/30 Updated pipeline. Removed last parameter --- readme.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/readme.md b/readme.md index 7e20ae3..88dfbd5 100644 --- a/readme.md +++ b/readme.md @@ -29,7 +29,7 @@ And then try out the model with just a couple lines of code: import torch from diffusers import VQDiffusionPipeline -pipeline = VQDiffusionPipeline.from_pretrained("microsoft/vq-diffusion-ithq", torch_dtype=torch.float16, revision="fp16") +pipeline = VQDiffusionPipeline.from_pretrained("microsoft/vq-diffusion-ithq", torch_dtype=torch.float16) pipeline = pipeline.to("cuda") image = pipeline("teddy bear playing in the pool").images[0]