frankleeeee
commited on
Commit
•
f8754d0
1
Parent(s):
36a01dc
Updated the sample code
Browse filesadded torch import to make it smoother to copy and run
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"
|