File size: 997 Bytes
7320bea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
---
license: mit
tags:
- diffusion
- concept-erasure
- stable-diffusion
- uce
- parachute
datasets:
- imagenet
language:
- en
pipeline_tag: text-to-image
---
# uce_parachute
This is a concept-erased Stable Diffusion model using the **Unlearning via Concept Erasure (UCE)** method to remove the concept **"Parachute"**.
## Method
Unlearning via Concept Erasure (UCE) modifies the model's internal representations to remove specific concept knowledge.
## Usage
```python
from diffusers import StableDiffusionPipeline
import torch
pipe = StableDiffusionPipeline.from_pretrained("ErasureResearch/uce_parachute", torch_dtype=torch.float16).to("cuda")
prompt = "a photo of a parachute"
image = pipe(prompt).images[0]
image.save("erased_parachute.png")
```
## Citation
If you use this model in your research, please cite:
```bibtex
@article{concept_erasure_2024,
title={Concept Erasure in Diffusion Models},
author={ErasureResearch Team},
journal={Proceedings of...},
year={2024}
}
```
|