pyproject.toml

[tool.poetry]
name = "sae-featureextraction"
version = "0.1.0"
description = ""
authors = ["ruben.haerle <ruben.haerle@aleph-alpha-ip.ai>"]
readme = "README.md"
package-mode = false

[tool.poetry.dependencies]
python = "^3.10"
torch = "^2.3.0"
transformers = {extras = ["sentencepiece"], version = "^4.43"}
plotly = "^5.22.0"
datasets = "^2.19.2"
lightning = "^2.2.5"
zstandard = "^0.22.0"
jupyter = "^1.0.0"
blobfile = "2.0.2"
transformer-lens = "1.9.1"
scikit-learn = "^1.5.0"
matplotlib = "^3.9.0"
graphviz = "^0.20.3"
google-api-python-client = "^2.135.0"
langdetect = "^1.0.9"
immutabledict = "^4.2.0"
huggingface-hub = {extras = ["cli"], version = "^0.23.4"}
# lm-eval = {path = "/nfs/scratch_2/ruben/lm-evaluation-harness", develop = true, extras=["ifeval", "math"]}
kaleido = "0.1.0"
# activation-additions = {path = "/nfs/scratch_2/ruben/activation_additions_hf-main"}
determined = "0.26.4"
tensorboard = "^2.17.0"
apex = "^0.9.10.dev0"
statsmodels = "^0.14.2"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"

[virtualenvs]
in-project = true
create = true
path = "./.venv"