pyproject.toml (16 lines of code) (raw):

[project] name = "sparse_autoencoder" description="Sparse autoencoder for GPT2" version = "0.1" authors = [{name = "OpenAI"}] dependencies = [ "blobfile == 2.0.2", "torch == 2.1.0", "transformer_lens == 1.9.1", ] readme = "README.md" [build-system] requires = ["setuptools>=64.0"] build-backend = "setuptools.build_meta" [tool.setuptools.packages.find] include = ["sparse_autoencoder*"]