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"
|