File size: 533 Bytes
8e1bfdc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
---
license: mit
language:
- en
---
# FastSAE weights
- `backbone/` — for `fastsae.models.Backbone.load(...)`
- `final/` — for `fastsae.models.SAE.load(...)`
## Usage
```python
from huggingface_hub import snapshot_download
from pathlib import Path
import fastsae
local = snapshot_download("YOUR_USER/fastsae-weights") # add token=... if private
backbone = fastsae.models.Backbone.load(Path(local)/"backbone")
sae = fastsae.models.SAE.load(Path(local)/"final")
wrapper = fastsae.models.SAEWrapper(backbone=backbone, sae=sae) |