Skip to content

Commit 8204415

Browse files
authored
Fix typo: torch_type -> torch_dtype (huggingface#972)
Fix typo: torch_type -> torch_dtype
1 parent 2fb8faf commit 8204415

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ precision while being roughly twice as fast and requiring half the amount of GPU
9090
```python
9191
from diffusers import StableDiffusionPipeline
9292

93-
pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_type=torch.float16, revision="fp16")
93+
pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, revision="fp16")
9494
pipe = pipe.to("cuda")
9595

9696
prompt = "a photo of an astronaut riding a horse on mars"

0 commit comments

Comments
 (0)