nielsr HF Staff commited on
Commit
4710ff2
·
verified ·
1 Parent(s): 55da75b

Add pipeline tag and update paper link/title

Browse files

This PR improves the model card by:

* Adding the `pipeline_tag: text-to-image` so the model can be discovered at https://huggingface.co/models?pipeline_tag=text-to-image.
* Updating the model's description to accurately refer to the paper [AlignGuard: Scalable Safety Alignment for Text-to-Image Generation](https://huggingface.co/papers/2412.10493).
* Aligning the display names of the project page and GitHub repository links with the paper's title.

Files changed (1) hide show
  1. README.md +6 -4
README.md CHANGED
@@ -1,16 +1,18 @@
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")
 
1
  ---
2
  library_name: diffusers
3
  license: mit
4
+ pipeline_tag: text-to-image
5
  ---
 
6
 
7
+ This is the official released checkpoint of **an aligned SD v2.1** for **AlignGuard: Scalable Safety Alignment for Text-to-Image Generation** ([paper](https://huggingface.co/papers/2412.10493)), designed to generate more safe images.
8
 
9
+ Our project page is [🏠AlignGuard HomePage](https://safetydpo.github.io/) and the GitHub repo is [⚙️AlignGuard GitHub](https://github.com/Visualignment/SafetyDPO) where we released all the code and the data.
10
+
11
+ In the future, we will release additional safe models.
12
 
13
  # Usage
14
  A simple use case of our model is:
15
+ ```python
16
  from diffusers import DiffusionPipeline
17
 
18
  pipe = DiffusionPipeline.from_pretrained("Visualignment/safe-stable-diffusion-v2-1")