Downloads · 30 days
0
safe-diffusion-guidance
basimazam/safe-diffusion-guidance
safe-diffusion-guidance is a machine learning model from basimazam. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
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…
Access
Public
Updated Aug 11, 2025
Repo size
90 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth180 MB · 100%
At a glance
- Access
- Public
- Created
- Aug 11, 2025
- Updated
- Aug 11, 2025
- SHA
- f4c05cb3
Model card
From the Hugging Face model README
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
python -m venv .venv && source .venv/bin/activate
pip install -r requirements.txt
- Created
- Aug 11, 2025
- Updated
- Aug 11, 2025