Commit
·
845609e
1
Parent(s):
b90003f
Updated the sample code (#58)
Browse files- Updated the sample code (f8754d06e54f1806a4eb097213eaa3bf3a668e40)
Co-authored-by: Frank Lee <[email protected]>
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"
|