rtliu commited on
Commit
55da75b
·
verified ·
1 Parent(s): f16b257

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -4
README.md CHANGED
@@ -1,4 +1,20 @@
1
- ---
2
- library_name: diffusers
3
- license: mit
4
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: diffusers
3
+ license: mit
4
+ ---
5
+ This is the official released checkpoint of **an aligned SD v2.1** of SafetyDPO: Scalable Safety Alignment for Text-to-Image Generation, designed to generate more safe images from our Safe-SD2.1.
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.
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-v2-1")
17
+
18
+ prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
19
+ image = pipe(prompt).images[0]
20
+ ```