patrickvonplaten commited on
Commit
fc392f6
1 Parent(s): 438a11a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -31,6 +31,7 @@ You can use this both with the [🧨Diffusers library](https://github.com/huggin
31
  ### Diffusers
32
  ```py
33
  from diffusers import StableDiffusionPipeline
 
34
 
35
  model_id = "runwayml/stable-diffusion-v1-5"
36
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16")
31
  ### Diffusers
32
  ```py
33
  from diffusers import StableDiffusionPipeline
34
+ import torch
35
 
36
  model_id = "runwayml/stable-diffusion-v1-5"
37
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16")