Skip to content

Commit 78f9c52

Browse files
patrickvonplatennatolambert
authored andcommitted
Improve README (#301)
1 parent fa7b906 commit 78f9c52

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

README.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ You need to accept the model license before downloading or using the Stable Diff
4646
```python
4747
# make sure you're logged in with `huggingface-cli login`
4848
from torch import autocast
49+
import torch
4950
from diffusers import StableDiffusionPipeline, LMSDiscreteScheduler
5051

5152
lms = LMSDiscreteScheduler(
@@ -77,6 +78,7 @@ The `StableDiffusionImg2ImgPipeline` lets you pass a text prompt and an initial
7778
```python
7879
from torch import autocast
7980
import requests
81+
import torch
8082
from PIL import Image
8183
from io import BytesIO
8284

@@ -116,6 +118,7 @@ The `StableDiffusionInpaintPipeline` lets you edit specific parts of an image by
116118
from io import BytesIO
117119

118120
from torch import autocast
121+
import torch
119122
import requests
120123
import PIL
121124

0 commit comments

Comments
 (0)