Skip to content

GitLab

Unverified Commit 82044153 authored by Pedro Cuenca's avatar Pedro Cuenca Committed by GitHub
Browse files

Fix typo: `torch_type` -> `torch_dtype` (#972) 

Fix typo: torch_type -> torch_dtype
parent 2fb8fafa
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
README.md
View file @ 82044153
......@@ -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"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment