Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
---
|
4 |
+
# SAeUron: Interpretable Concept Unlearning in Diffusion Models with Sparse Autoencoders
|
5 |
+
|
6 |
+
[Paper on arXiv](https://arxiv.org/abs/2501.18052) |
|
7 |
+
[GitHub repo](https://github.com/cywinski/SAeUron)
|
8 |
+
|
9 |
+
The repository contains Sparse Autoencoders trained in our work for blocks `up.1.1` and `up.1.2`.
|
10 |
+
|
11 |
+
After cloning our GitHub repo, you can use them as follows:
|
12 |
+
```python
|
13 |
+
from SAE.sae import Sae
|
14 |
+
|
15 |
+
device = "cuda"
|
16 |
+
hookpoint = "unet.up_blocks.1.attentions.1"
|
17 |
+
|
18 |
+
sae = Sae.load_from_hub("bcywinski/SAeUron", hookpoint=hookpoint, device=device)
|
19 |
+
```
|