NYU-DICE-Lab/circumventing-concept-erasure

Circumventing Concept Erasure Methods For Text-to-Image Models

Project Website | Arxiv Preprint

Starting Guide

To get started, create a new conda environment and install the required packages:

conda create -f environment.yml

Optional: Get used to the 🤗Textual Inversion GitHub repository.

Training Guide

We provide Concept Inversion code for the following methods:

• Erased Stable Diffusion (ESD) in the /esd folder.
• Forget-Me-Not (FMN) in the /fmn folder.
• Selective Amnesia (SA) in the /sa folder.
• Negative Prompt (NP) in the /np folder.
• Safe Latent Diffusion (SLD) in the /sld folder.

Please refer to each folder for additional instructions, which is designed to be self-contained.

Acknowledgements

We would like to thank the authors of 🤗Difusersfor relesing their helpful codebases.