Update README.md

#54
by staxjp - opened
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -38,11 +38,12 @@ This `stable-diffusion-2-1` model is fine-tuned from [stable-diffusion-2](https:
38
  Using the [🤗's Diffusers library](https://github.com/huggingface/diffusers) to run Stable Diffusion 2 in a simple and efficient manner.
39
 
40
  ```bash
41
- pip install diffusers transformers accelerate scipy safetensors
42
  ```
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"
 
38
  Using the [🤗's Diffusers library](https://github.com/huggingface/diffusers) to run Stable Diffusion 2 in a simple and efficient manner.
39
 
40
  ```bash
41
+ pip install diffusers transformers accelerate scipy safetensors torch
42
  ```
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"