Update README.md
#74
by
ZuraGuerra
- opened
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"
|