harishp commited on
Commit
e3888b9
1 Parent(s): ac79876

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -29,7 +29,7 @@ You can use the pipeline like so:
29
  from diffusers import DiffusionPipeline
30
  import torch
31
 
32
- pipeline = DiffusionPipeline.from_pretrained("Warlord-K/BKSDM-Base-95K", torch_dtype=torch.float16)
33
  prompt = "Portrait of a pretty girl"
34
  image = pipeline(prompt).images[0]
35
  image.save("my_image.png")
 
29
  from diffusers import DiffusionPipeline
30
  import torch
31
 
32
+ pipeline = DiffusionPipeline.from_pretrained("segmind/small-sd", torch_dtype=torch.float16)
33
  prompt = "Portrait of a pretty girl"
34
  image = pipeline(prompt).images[0]
35
  image.save("my_image.png")