Text-to-Image
Diffusers
Safetensors
StableDiffusionPipeline
stable-diffusion
Inference Endpoints
patrickvonplaten frankleeeee commited on
Commit
845609e
1 Parent(s): b90003f

Updated the sample code (#58)

Browse files

- Updated the sample code (f8754d06e54f1806a4eb097213eaa3bf3a668e40)


Co-authored-by: Frank Lee <[email protected]>

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"