metadata
license: mit
tags:
- diffusion
- concept-erasure
- stable-diffusion
- uce
- airliner
datasets:
- imagenet
language:
- en
pipeline_tag: text-to-image
uce_airliner
This is a concept-erased Stable Diffusion model using the Unlearning via Concept Erasure (UCE) method to remove the concept "Airliner".
Method
Unlearning via Concept Erasure (UCE) modifies the model's internal representations to remove specific concept knowledge.
Usage
from diffusers import StableDiffusionPipeline
import torch
pipe = StableDiffusionPipeline.from_pretrained("ErasureResearch/uce_airliner", torch_dtype=torch.float16).to("cuda")
prompt = "a photo of a airliner"
image = pipe(prompt).images[0]
image.save("erased_airliner.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}
}