avinchintha commited on
Commit
848e43e
1 Parent(s): 07753ec

import torch missing in stable-diffusion-2 usage snippet

Browse files

Just imported the torch package in the example usage code snippet so it can be copied as-is and run without errors.

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