ItsMaxNorm's picture
Release empty-positive variant (full SD pipeline)
9a7a271 verified
|
raw
history blame
832 Bytes
metadata
library_name: diffusers
pipeline_tag: text-to-image
license: mit
base_model: runwayml/stable-diffusion-v1-5
tags:
  - stable-diffusion
  - safety
  - grpo
  - steering-reward
  - safediffusion-r1

SafeDiffusion-R1 — empty-positive variant

Empty-positive ablation (only 3 negative anchors, no safe target; geneval_negative_steringreward_8gpus_empty_pos @ epoch 280).

Quick load

from diffusers import StableDiffusionPipeline
import torch
pipe = StableDiffusionPipeline.from_pretrained(
    "ItsMaxNorm/SafeDiffusion-R1", subfolder="empty-positive", torch_dtype=torch.float16
).to("cuda")
img = pipe("a photo of a cat sleeping on a couch").images[0]
img.save("out.png")

See the main repo card for the method, headline numbers, training recipe, and citation: https://huggingface.co/ItsMaxNorm/SafeDiffusion-R1.