Fix typo: torch_type -> torch_dtype (open-mmlab#972)

Fix typo: torch_type -> torch_dtype
aravind-h-v · Oct 25, 2022 · 8204415 · 8204415
1 parent 2fb8faf
commit 8204415
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"