Text-to-Image
Diffusers
Safetensors
StableDiffusionPipeline
stable-diffusion
Inference Endpoints
frankleeeee commited on
Commit
f8754d0
1 Parent(s): 36a01dc

Updated the sample code

Browse files

added torch import to make it smoother to copy and run

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"