From c6fdceb419230aa946617f23e4440106d6d7aa57 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Tue, 25 Oct 2022 09:13:26 +0200 Subject: [PATCH] Fix typo: torch_type -> torch_dtype --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 1dc1d7838a5d..a5f47dfcb032 100644 --- a/README.md +++ b/README.md @@ -90,7 +90,7 @@ precision while being roughly twice as fast and requiring half the amount of GPU ```python from diffusers import StableDiffusionPipeline -pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_type=torch.float16, revision="fp16") +pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, revision="fp16") pipe = pipe.to("cuda") prompt = "a photo of an astronaut riding a horse on mars"