Text-to-Image
Diffusers
Safetensors
English
StableDiffusionPipeline
diffusion
concept-erasure
stable-diffusion
esdx
tench
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("DiffusionConceptErasure/esdx_tench", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]esdx_tench
This is a concept-erased Stable Diffusion model using the Exact Source Distillation (ESD-X) method to remove the concept "Tench".
Method
Exact Source Distillation (ESD-X) erases concepts by distilling knowledge while excluding specific concept representations.
Usage
from diffusers import StableDiffusionPipeline
import torch
pipe = StableDiffusionPipeline.from_pretrained("ErasureResearch/esdx_tench", torch_dtype=torch.float16).to("cuda")
prompt = "a photo of a tench"
image = pipe(prompt).images[0]
image.save("erased_tench.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}
}
- Downloads last month
- 1