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

Update ReadMe.md with working code example

Browse files

Updating ReadMe.md code example with `import torch` for a working code example.

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"