AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Fix typo: torch_type -> torch_dtype (#972)

Browse Source 
Fix typo: torch_type -> torch_dtype
This commit is contained in:
Pedro Cuenca
2022-10-25 13:05:44 +02:00
committed by GitHub
parent 2fb8fafa4b
commit 82044153df
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -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"