Skip to content

Commit

Permalink
feat: use TransformerLens 2 (#214)
Browse files Browse the repository at this point in the history
* Updated pyproject.toml to use TL ^2.0, and to use fork of sae-vis that also uses TL ^2.0

* Removed reliance on sae-vis

* Removed neuronpedia tutorial

* Added error handling for view operation

* Corrected formatting
  • Loading branch information
curt-tigges authored Jul 9, 2024
1 parent 697dd5f commit 526e736
Show file tree
Hide file tree
Showing 8 changed files with 5,429 additions and 1,168 deletions.
3 changes: 1 addition & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ include = ["pretrained_saes.yaml"]

[tool.poetry.dependencies]
python = "^3.10"
transformer-lens = "^1.14.0"
transformer-lens = "^2.0.0"
transformers = "^4.38.1"
plotly = "^5.19.0"
plotly-express = "^0.4.1"
Expand All @@ -18,7 +18,6 @@ matplotlib-inline = "^0.1.6"
datasets = "^2.17.1"
babe = "^0.0.7"
nltk = "^3.8.1"
sae-vis = "^0.2.18"
safetensors = "^0.4.2"
typer = "^0.12.3"
mamba-lens = { version = "^0.0.4", optional = true }
Expand Down
Loading

0 comments on commit 526e736

Please sign in to comment.