Modelpublic

DiffusionConceptErasure/esdu_golf_ball

sourceHugging Facemitupdated 1y agoView on Hugging Face

0likes2downloads

Model Card

esdugolfball

This is a concept-erased Stable Diffusion model using the Unconstrained Source Distillation (ESD-U) method to remove the concept "Golf Ball".

Method

Unconstrained Source Distillation (ESD-U) performs unconstrained distillation to remove concept information.

Usage

python

from diffusers import StableDiffusionPipeline
import torch

pipe = StableDiffusionPipeline.from_pretrained("ErasureResearch/esdu_golf_ball", torch_dtype=torch.float16).to("cuda")
prompt = "a photo of a golf_ball"
image = pipe(prompt).images[0]
image.save("erased_golf_ball.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}
}