File size: 445 Bytes
fbfadc4
febc264
fbfadc4
 
 
 
febc264
fbfadc4
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
# Safe Diffusion Guidance (SDG)

Custom Diffusers pipeline that applies a mid-UNet safety classifier as guidance during denoising.
- Plug-and-play: works with any Stable Diffusion checkpoint (e.g., SD 1.5).
- No retraining needed; classifier runs on mid-UNet features.
- Tunable: `safety_scale`, `mid_fraction`, `safe_class_index`.

## Install
```bash

python -m venv .venv && source .venv/bin/activate

pip install -r requirements.txt