import missing module in example
#16
by
HuggingDavid
- opened
README.md
CHANGED
@@ -46,6 +46,7 @@ Running the pipeline (if you don't swap the scheduler it will run with the defau
|
|
46 |
|
47 |
```python
|
48 |
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
|
|
|
49 |
|
50 |
model_id = "stabilityai/stable-diffusion-2"
|
51 |
|
|
|
46 |
|
47 |
```python
|
48 |
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
|
49 |
+
import torch
|
50 |
|
51 |
model_id = "stabilityai/stable-diffusion-2"
|
52 |
|