DiffusionConceptErasure/esdx_tench

1---2license: mit3tags:4- diffusion5- concept-erasure6- stable-diffusion7- esdx8- tench9datasets:10- imagenet11language:12- en13pipeline_tag: text-to-image14---15 16# esdx_tench17 18This is a concept-erased Stable Diffusion model using the **Exact Source Distillation (ESD-X)** method to remove the concept **"Tench"**.19 20## Method21 22Exact Source Distillation (ESD-X) erases concepts by distilling knowledge while excluding specific concept representations.23 24## Usage25 26```python27from diffusers import StableDiffusionPipeline28import torch29 30pipe = StableDiffusionPipeline.from_pretrained("ErasureResearch/esdx_tench", torch_dtype=torch.float16).to("cuda")31prompt = "a photo of a tench"32image = pipe(prompt).images[0]33image.save("erased_tench.png")34```35 36## Citation37 38If you use this model in your research, please cite:39 40```bibtex41@article{concept_erasure_2024,42  title={Concept Erasure in Diffusion Models},43  author={ErasureResearch Team},44  journal={Proceedings of...},45  year={2024}46}47```48