feature-interp/setup.py at master · GeorgeMLP/feature-interp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
from setuptools import setup, find_packages


setup(
    name="featureinterp",
    packages=find_packages(),
    version="0.0.1",
    author="Samuel Pfrommer",
    install_requires=[
        "httpx>=0.22",
        "scikit-learn",
        "boostedblob>=0.13.0",
        "tiktoken",
        "blobfile",
        "numpy",
        "pytest",
        "orjson",
        "together",
        "dacite",
        "torch",
        "torchvision",
        "einops",
        "jaxtyping",
        "datasets",
        "transformers",
        "tqdm",
        "wandb",
        "sae_lens",
        "transformer_lens",
        "sentence_transformers",
        "info-nce-pytorch",
        "pdbpp",
        "colorama",
        "bitsandbytes",
        "json-repair",
        "google-generativeai",
    ],
    url="",
    description="",
    python_requires='>=3.9',
)