Text-to-Image
Diffusers
Safetensors
stable-diffusion
safety
safe-generation
grpo
steering-reward
nsfw-removal
safediffusion-r1
Instructions to use ItsMaxNorm/SafeDiffusion-R1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use ItsMaxNorm/SafeDiffusion-R1 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("ItsMaxNorm/SafeDiffusion-R1", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps
- Draw Things
- DiffusionBee
Update README.md
Browse files
README.md
CHANGED
|
@@ -86,10 +86,13 @@ works directly against this Hub release:
|
|
| 86 |
## Citation
|
| 87 |
|
| 88 |
```bibtex
|
| 89 |
-
@
|
| 90 |
-
|
| 91 |
-
|
| 92 |
-
|
| 93 |
-
|
|
|
|
|
|
|
|
|
|
| 94 |
}
|
| 95 |
```
|
|
|
|
| 86 |
## Citation
|
| 87 |
|
| 88 |
```bibtex
|
| 89 |
+
@misc{kumar2026safediffusionr1,
|
| 90 |
+
title={SafeDiffusion-R1: Online Reward Steering for Safe Diffusion Post-Training},
|
| 91 |
+
author={Komal Kumar and Ankan Deria and Abhishek Basu and Fahad Shamshad and Hisham Cholakkal and Karthik Nandakumar},
|
| 92 |
+
year={2026},
|
| 93 |
+
eprint={2605.18719},
|
| 94 |
+
archivePrefix={arXiv},
|
| 95 |
+
primaryClass={cs.CV},
|
| 96 |
+
url={https://arxiv.org/abs/2605.18719},
|
| 97 |
}
|
| 98 |
```
|