91 lines
5.6 KiB
Plaintext
91 lines
5.6 KiB
Plaintext
<!--Copyright 2022 The HuggingFace Team. All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
|
|
the License. You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
|
|
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
|
specific language governing permissions and limitations under the License.
|
|
-->
|
|
|
|
# Safe Stable Diffusion
|
|
|
|
Safe Stable Diffusion was proposed in [Safe Latent Diffusion: Mitigating Inappropriate Degeneration in Diffusion Models](https://arxiv.org/abs/2211.05105) and mitigates the well known issue that models like Stable Diffusion that are trained on unfiltered, web-crawled datasets tend to suffer from inappropriate degeneration. For instance Stable Diffusion may unexpectedly generate nudity, violence, images depicting self-harm, or otherwise offensive content.
|
|
Safe Stable Diffusion is an extension to the Stable Diffusion that drastically reduces content like this.
|
|
|
|
The abstract of the paper is the following:
|
|
|
|
*Text-conditioned image generation models have recently achieved astonishing results in image quality and text alignment and are consequently employed in a fast-growing number of applications. Since they are highly data-driven, relying on billion-sized datasets randomly scraped from the internet, they also suffer, as we demonstrate, from degenerated and biased human behavior. In turn, they may even reinforce such biases. To help combat these undesired side effects, we present safe latent diffusion (SLD). Specifically, to measure the inappropriate degeneration due to unfiltered and imbalanced training sets, we establish a novel image generation test bed-inappropriate image prompts (I2P)-containing dedicated, real-world image-to-text prompts covering concepts such as nudity and violence. As our exhaustive empirical evaluation demonstrates, the introduced SLD removes and suppresses inappropriate image parts during the diffusion process, with no additional training required and no adverse effect on overall image quality or text alignment.*
|
|
|
|
|
|
*Overview*:
|
|
|
|
| Pipeline | Tasks | Colab | Demo
|
|
|---|---|:---:|:---:|
|
|
| [pipeline_stable_diffusion_safe.py](https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion_safe/pipeline_stable_diffusion_safe.py) | *Text-to-Image Generation* | [](https://colab.research.google.com/github/ml-research/safe-latent-diffusion/blob/main/examples/Safe%20Latent%20Diffusion.ipynb) | -
|
|
|
|
## Tips
|
|
|
|
- Safe Stable Diffusion may also be used with weights of [Stable Diffusion](./api/pipelines/stable_diffusion).
|
|
|
|
### Run Safe Stable Diffusion
|
|
|
|
Safe Stable Diffusion can be tested very easily with the [`StableDiffusionPipelineSafe`], and the `"AIML-TUDA/stable-diffusion-safe"` checkpoint exactly in the same way it is shown in the [Conditional Image Generation Guide](./using-diffusers/conditional_image_generation).
|
|
|
|
### Interacting with the Safety Concept
|
|
|
|
To check and edit the currently used safety concept, use the `safety_concept` property of [`StableDiffusionPipelineSafe`]
|
|
```python
|
|
>>> from diffusers import StableDiffusionPipelineSafe
|
|
|
|
>>> pipeline = StableDiffusionPipelineSafe.from_pretrained("AIML-TUDA/stable-diffusion-safe")
|
|
>>> pipeline.safety_concept
|
|
```
|
|
For each image generation the active concept is also contained in [`StableDiffusionSafePipelineOutput`].
|
|
|
|
### Using pre-defined safety configurations
|
|
|
|
You may use the 4 configurations defined in the [Safe Latent Diffusion paper](https://arxiv.org/abs/2211.05105) as follows:
|
|
|
|
```python
|
|
>>> from diffusers import StableDiffusionPipelineSafe
|
|
>>> from diffusers.pipelines.stable_diffusion_safe import SafetyConfig
|
|
|
|
>>> pipeline = StableDiffusionPipelineSafe.from_pretrained("AIML-TUDA/stable-diffusion-safe")
|
|
>>> prompt = "the four horsewomen of the apocalypse, painting by tom of finland, gaston bussiere, craig mullins, j. c. leyendecker"
|
|
>>> out = pipeline(prompt=prompt, **SafetyConfig.MAX)
|
|
```
|
|
|
|
The following configurations are available: `SafetyConfig.WEAK`, `SafetyConfig.MEDIUM`, `SafetyConfig.STRONg`, and `SafetyConfig.MAX`.
|
|
|
|
### How to load and use different schedulers.
|
|
|
|
The safe stable diffusion pipeline uses [`PNDMScheduler`] scheduler by default. But `diffusers` provides many other schedulers that can be used with the stable diffusion pipeline such as [`DDIMScheduler`], [`LMSDiscreteScheduler`], [`EulerDiscreteScheduler`], [`EulerAncestralDiscreteScheduler`] etc.
|
|
To use a different scheduler, you can either change it via the [`ConfigMixin.from_config`] method or pass the `scheduler` argument to the `from_pretrained` method of the pipeline. For example, to use the [`EulerDiscreteScheduler`], you can do the following:
|
|
|
|
```python
|
|
>>> from diffusers import StableDiffusionPipelineSafe, EulerDiscreteScheduler
|
|
|
|
>>> pipeline = StableDiffusionPipelineSafe.from_pretrained("AIML-TUDA/stable-diffusion-safe")
|
|
>>> pipeline.scheduler = EulerDiscreteScheduler.from_config(pipeline.scheduler.config)
|
|
|
|
>>> # or
|
|
>>> euler_scheduler = EulerDiscreteScheduler.from_pretrained("AIML-TUDA/stable-diffusion-safe", subfolder="scheduler")
|
|
>>> pipeline = StableDiffusionPipelineSafe.from_pretrained(
|
|
... "AIML-TUDA/stable-diffusion-safe", scheduler=euler_scheduler
|
|
... )
|
|
```
|
|
|
|
|
|
## StableDiffusionSafePipelineOutput
|
|
[[autodoc]] pipelines.stable_diffusion_safe.StableDiffusionSafePipelineOutput
|
|
|
|
## StableDiffusionPipelineSafe
|
|
[[autodoc]] StableDiffusionPipelineSafe
|
|
- __call__
|
|
- enable_attention_slicing
|
|
- disable_attention_slicing
|
|
|