avinchintha
commited on
Commit
•
848e43e
1
Parent(s):
07753ec
import torch missing in stable-diffusion-2 usage snippet
Browse filesJust imported the torch package in the example usage code snippet so it can be copied as-is and run without errors.
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"
|