ZuraGuerra
commited on
Commit
•
866ec42
1
Parent(s):
1e128c8
Update README.md
Browse filesAdd missing import from example
README.md
CHANGED
@@ -45,6 +45,7 @@ pip install diffusers transformers accelerate scipy safetensors
|
|
45 |
Running the pipeline (if you don't swap the scheduler it will run with the default DDIM, in this example we are swapping it to EulerDiscreteScheduler):
|
46 |
|
47 |
```python
|
|
|
48 |
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
|
49 |
|
50 |
model_id = "stabilityai/stable-diffusion-2"
|
|
|
45 |
Running the pipeline (if you don't swap the scheduler it will run with the default DDIM, in this example we are swapping it to EulerDiscreteScheduler):
|
46 |
|
47 |
```python
|
48 |
+
import torch
|
49 |
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
|
50 |
|
51 |
model_id = "stabilityai/stable-diffusion-2"
|