stabilityai
/

stable-diffusion-2-1

StableDiffusionPipeline

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

staxjp commited on May 3, 2023

Commit

f615a70

•

1 Parent(s): 36a01dc

Update README.md

It requires torch to be imported as well.

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -38,11 +38,12 @@ This `stable-diffusion-2-1` model is fine-tuned from [stable-diffusion-2](https:
 Using the [🤗's Diffusers library](https://github.com/huggingface/diffusers) to run Stable Diffusion 2 in a simple and efficient manner.
 ```bash
-pip install diffusers transformers accelerate scipy safetensors
 ```
 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):
 ```python
 from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 model_id = "stabilityai/stable-diffusion-2-1"

 Using the [🤗's Diffusers library](https://github.com/huggingface/diffusers) to run Stable Diffusion 2 in a simple and efficient manner.
 ```bash
+pip install diffusers transformers accelerate scipy safetensors torch
 ```
 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):
 ```python
+import torch
 from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 model_id = "stabilityai/stable-diffusion-2-1"