Modelpublic

DiffusionConceptErasure/ga_english_springer_spaniel

sourceHugging Facemitupdated 1y agoView on Hugging Face

0likes2downloads

Model Card

gaenglishspringer_spaniel

This is a concept-erased Stable Diffusion model using the Gradient Ascent (GA) method to remove the concept "English Springer Spaniel".

Method

Gradient Ascent (GA) actively pushes the model away from generating the target concept.

Usage

python

from diffusers import StableDiffusionPipeline
import torch

pipe = StableDiffusionPipeline.from_pretrained("ErasureResearch/ga_english_springer_spaniel", torch_dtype=torch.float16).to("cuda")
prompt = "a photo of a english_springer_spaniel"
image = pipe(prompt).images[0]
image.save("erased_english_springer_spaniel.png")

Citation

If you use this model in your research, please cite:

bibtex

@article{concept_erasure_2024,
  title={Concept Erasure in Diffusion Models},
  author={ErasureResearch Team},
  journal={Proceedings of...},
  year={2024}
}