rtliu commited on
Commit
1cc4fee
·
verified ·
1 Parent(s): f940ff3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -2
README.md CHANGED
@@ -2,8 +2,19 @@
2
  library_name: diffusers
3
  license: mit
4
  ---
5
- This is the official released checkpoint of **an algined Stable Diffusion v1.5** of SafetyDPO: Scalable Safety Alignment for Text-to-Image Generation, designed to generate safe images from text.
6
 
7
  Our project page is [🏠SafetyDPO HomePage](https://safetydpo.github.io/) and the GitHub repo is [⚙️SafetyDPO GitHub](https://github.com/Visualignment/SafetyDPO) where we released all the code and the data.
8
 
9
- In the future, we will release additional safe models, including the safe-SDXL.
 
 
 
 
 
 
 
 
 
 
 
 
2
  library_name: diffusers
3
  license: mit
4
  ---
5
+ This is the official released checkpoint of **an algined Stable Diffusion v1.5** of SafetyDPO: Scalable Safety Alignment for Text-to-Image Generation, designed to generate more safe images from our Safe-SD1.5.
6
 
7
  Our project page is [🏠SafetyDPO HomePage](https://safetydpo.github.io/) and the GitHub repo is [⚙️SafetyDPO GitHub](https://github.com/Visualignment/SafetyDPO) where we released all the code and the data.
8
 
9
+ In the future, we will release additional safe models, including the safe-SDXL.
10
+
11
+ # Usage
12
+ A simple use case of our model is:
13
+ ```
14
+ from diffusers import DiffusionPipeline
15
+
16
+ pipe = DiffusionPipeline.from_pretrained("Visualignment/safe-stable-diffusion-v1-5")
17
+
18
+ prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
19
+ image = pipe(prompt).images[0]
20
+ ```