|
|
--- |
|
|
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) |
