nielsr HF Staff commited on
Commit
c2a2ec6
·
verified ·
1 Parent(s): 1cc4fee

Update model card with correct paper details and pipeline tag

Browse files

This PR improves the model card by:
- Updating the introductory text to correctly reference and link the official Hugging Face Papers page for [AlignGuard: Scalable Safety Alignment for Text-to-Image Generation](https://huggingface.co/papers/2412.10493).
- Adding the `pipeline_tag: text-to-image` to the metadata, ensuring the model is correctly categorized and discoverable on the Hub at https://huggingface.co/models?pipeline_tag=text-to-image.

Files changed (1) hide show
  1. README.md +6 -4
README.md CHANGED
@@ -1,12 +1,14 @@
1
  ---
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:
@@ -17,4 +19,4 @@ pipe = DiffusionPipeline.from_pretrained("Visualignment/safe-stable-diffusion-v1
17
 
18
  prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
19
  image = pipe(prompt).images[0]
20
- ```
 
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 Stable Diffusion v1.5** based on the method presented in [AlignGuard: Scalable Safety Alignment for Text-to-Image Generation](https://huggingface.co/papers/2412.10493), designed to generate safer images.
8
 
9
+ 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.
10
+
11
+ In the future, we will release additional safe models, including the safe-SDXL.
12
 
13
  # Usage
14
  A simple use case of our model is:
 
19
 
20
  prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
21
  image = pipe(prompt).images[0]
22
+ ```