| | --- |
| | library_name: diffusers |
| | license: mit |
| | --- |
| | 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. |
| |
|
| | 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. |
| |
|
| | In the future, we will release additional safe models. |
| |
|
| | # Usage |
| | A simple use case of our model is: |
| | ``` |
| | from diffusers import DiffusionPipeline |
| | |
| | pipe = DiffusionPipeline.from_pretrained("Visualignment/safe-stable-diffusion-v2-1") |
| | |
| | prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" |
| | image = pipe(prompt).images[0] |
| | ``` |
