the example code miss import torch
Browse filesadd import torch to handle error:
``` pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
NameError: name 'torch' is not defined
```
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-base"
|
51 |
|
|
|
46 |
|
47 |
```python
|
48 |
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
|
49 |
+
import torch
|
50 |
|
51 |
model_id = "stabilityai/stable-diffusion-2-base"
|
52 |
|