We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
torch_type
torch_dtype
1 parent 2fb8faf commit 8204415Copy full SHA for 8204415
README.md
@@ -90,7 +90,7 @@ precision while being roughly twice as fast and requiring half the amount of GPU
90
```python
91
from diffusers import StableDiffusionPipeline
92
93
-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")
94
pipe = pipe.to("cuda")
95
96
prompt = "a photo of an astronaut riding a horse on mars"
0 commit comments