Update README.md code to import `torch`
Browse filesThe sample code snippet doesn't run by itself because `torch` would be undefined in
```python
StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
```
README.md
CHANGED
@@ -46,6 +46,7 @@ Running the pipeline (if you don't swap the scheduler it will run with the defau
|
|
46 |
|
47 |
```python
|
48 |
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
|
|
|
49 |
|
50 |
model_id = "stabilityai/stable-diffusion-2"
|
51 |
|
|
|
46 |
|
47 |
```python
|
48 |
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
|
49 |
+
import torch
|
50 |
|
51 |
model_id = "stabilityai/stable-diffusion-2"
|
52 |
|