metadata
license: mit
tags:
- diffusion
- concept-erasure
- stable-diffusion
- esdu
- picasso
datasets:
- imagenet
language:
- en
pipeline_tag: text-to-image
esdu_picasso
This is a concept-erased Stable Diffusion model using the Unconstrained Source Distillation (ESD-U) method to remove the concept "Picasso".
Method
Unconstrained Source Distillation (ESD-U) performs unconstrained distillation to remove concept information.
Usage
from diffusers import StableDiffusionPipeline
import torch
pipe = StableDiffusionPipeline.from_pretrained("ErasureResearch/esdu_picasso", torch_dtype=torch.float16).to("cuda")
prompt = "a photo of a picasso"
image = pipe(prompt).images[0]
image.save("erased_picasso.png")
Citation
If you use this model in your research, please cite:
@article{concept_erasure_2024,
title={Concept Erasure in Diffusion Models},
author={ErasureResearch Team},
journal={Proceedings of...},
year={2024}
}