Text-to-Image
Diffusers
Safetensors
StableDiffusionPipeline
stable-diffusion
Inference Endpoints

Update ReadMe.md with working code example

#36
by mesnarj - opened
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -43,6 +43,7 @@ pip install diffusers transformers accelerate scipy safetensors
43
  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 DPMSolverMultistepScheduler):
44
 
45
  ```python
 
46
  from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
47
 
48
  model_id = "stabilityai/stable-diffusion-2-1"
 
43
  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 DPMSolverMultistepScheduler):
44
 
45
  ```python
46
+ import torch
47
  from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
48
 
49
  model_id = "stabilityai/stable-diffusion-2-1"