diff --git a/README.md b/README.md index 1dc1d7838a..a5f47dfcb0 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"