Skip to content

Commit 2f0fcf4

Browse files
authored
Add missing import (#979)
1 parent cc43608 commit 2f0fcf4

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,7 @@ We recommend using the model in [half-precision (`fp16`)](https://pytorch.org/bl
8888
precision while being roughly twice as fast and requiring half the amount of GPU RAM.
8989

9090
```python
91+
import torch
9192
from diffusers import StableDiffusionPipeline
9293

9394
pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, revision="fp16")

0 commit comments

Comments
 (0)