From 52e8fdb8aec7c3676f2e6987291d16167aad9f4e Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 17 Oct 2022 15:25:04 +0200 Subject: [PATCH] Update README.md --- examples/community/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/community/README.md b/examples/community/README.md index 00ca0db4..cc9627c0 100644 --- a/examples/community/README.md +++ b/examples/community/README.md @@ -133,7 +133,7 @@ def download_image(url): response = requests.get(url) return PIL.Image.open(BytesIO(response.content)).convert("RGB") -pipe = DiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", custom_pipeline="stable_diffusion_mega", dtype=torch.float16, revision="fp16") +pipe = DiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", custom_pipeline="stable_diffusion_mega", torch_dtype=torch.float16, revision="fp16") pipe.to("cuda") pipe.enable_attention_slicing()