bcywinski commited on
Commit
09e2610
·
verified ·
1 Parent(s): a469268

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -0
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
+ ```