yotmic commited on
Commit
0074425
1 Parent(s): 0904434

the example code miss import torch

Browse files

add 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
```

Files changed (1) hide show
  1. README.md +1 -0
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