the example code miss import torch (#4)
Browse files- the example code miss import torch (0074425cd97fcde7d97323b12a6277a70ddd373e)
Co-authored-by: yotam michael <yotmic@users.noreply.huggingface.co>
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-base"
|
51 |
|
|
|
46 |
|
47 |
```python
|
48 |
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
|
49 |
+
import torch
|
50 |
|
51 |
model_id = "stabilityai/stable-diffusion-2-base"
|
52 |
|