fixed the bug of import torch

#33
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -44,6 +44,7 @@ pip install --upgrade git+https://github.com/huggingface/diffusers.git transform
44
  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):
45
 
46
  ```python
 
47
  from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
48
 
49
  model_id = "stabilityai/stable-diffusion-2"
44
  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):
45
 
46
  ```python
47
+ import torch
48
  from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
49
 
50
  model_id = "stabilityai/stable-diffusion-2"