Update ReadMe.md with working code example
#36
by
mesnarj
- opened
README.md
CHANGED
@@ -43,6 +43,7 @@ pip install diffusers transformers accelerate scipy safetensors
|
|
43 |
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 DPMSolverMultistepScheduler):
|
44 |
|
45 |
```python
|
|
|
46 |
from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
|
47 |
|
48 |
model_id = "stabilityai/stable-diffusion-2-1"
|
|
|
43 |
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 DPMSolverMultistepScheduler):
|
44 |
|
45 |
```python
|
46 |
+
import torch
|
47 |
from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
|
48 |
|
49 |
model_id = "stabilityai/stable-diffusion-2-1"
|