-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
119 lines (118 loc) · 3.34 KB
/
setup.py
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
from setuptools import setup, find_packages
setup(
name="info-theory-experiments",
version="0.1",
packages=find_packages(),
install_requires=[
"accelerate==0.33.0",
"aiohappyeyeballs==2.3.4",
"aiohttp==3.10.0",
"aiosignal==1.3.1",
"asttokens==2.4.1",
"async-timeout==4.0.3",
"attrs==23.2.0",
"beartype==0.14.1",
"better-abc==0.0.3",
"certifi==2024.7.4",
"charset-normalizer==3.3.2",
"click==8.1.7",
"comm==0.2.2",
"contourpy==1.2.1",
"cycler==0.12.1",
"datasets==2.20.0",
"debugpy==1.8.2",
"decorator==5.1.1",
"dill==0.3.8",
"docker-pycreds==0.4.0",
"einops==0.8.0",
"exceptiongroup==1.2.2",
"executing==2.0.1",
"fancy-einsum==0.0.3",
"filelock==3.15.4",
"fonttools==4.53.1",
"frozenlist==1.4.1",
"fsspec==2024.5.0",
"gitdb==4.0.11",
"GitPython==3.1.43",
"huggingface-hub==0.24.5",
"idna==3.7",
"ipykernel==6.29.5",
"ipython==8.26.0",
"jaxtyping==0.2.33",
"jedi==0.19.1",
"Jinja2==3.1.4",
"jupyter_client==8.6.2",
"jupyter_core==5.7.2",
"kiwisolver==1.4.5",
"markdown-it-py==3.0.0",
"MarkupSafe==2.1.5",
"matplotlib==3.9.1",
"matplotlib-inline==0.1.7",
"mdurl==0.1.2",
"mpmath==1.3.0",
"multidict==6.0.5",
"multiprocess==0.70.16",
"nest-asyncio==1.6.0",
"networkx==3.3",
"numpy==1.26.4",
"nvidia-cublas-cu12==12.1.3.1",
"nvidia-cuda-cupti-cu12==12.1.105",
"nvidia-cuda-nvrtc-cu12==12.1.105",
"nvidia-cuda-runtime-cu12==12.1.105",
"nvidia-cudnn-cu12==9.1.0.70",
"nvidia-cufft-cu12==11.0.2.54",
"nvidia-curand-cu12==10.3.2.106",
"nvidia-cusolver-cu12==11.4.5.107",
"nvidia-cusparse-cu12==12.1.0.106",
"nvidia-nccl-cu12==2.20.5",
"nvidia-nvjitlink-cu12==12.5.82",
"nvidia-nvtx-cu12==12.1.105",
"packaging==24.1",
"pandas==2.2.2",
"parso==0.8.4",
"pexpect==4.9.0",
"pillow==10.4.0",
"platformdirs==4.2.2",
"prompt_toolkit==3.0.47",
"protobuf==5.27.3",
"psutil==6.0.0",
"ptyprocess==0.7.0",
"pure_eval==0.2.3",
"pyarrow==17.0.0",
"pyarrow-hotfix==0.6",
"Pygments==2.18.0",
"pyparsing==3.1.2",
"python-dateutil==2.9.0.post0",
"pytz==2024.1",
"PyYAML==6.0.1",
"pyzmq==26.0.3",
"regex==2024.7.24",
"requests==2.32.3",
"rich==13.7.1",
"safetensors==0.4.3",
"scipy==1.14.0",
"sentencepiece==0.2.0",
"sentry-sdk==2.12.0",
"setproctitle==1.3.3",
"six==1.16.0",
"smmap==5.0.1",
"stack-data==0.6.3",
"sympy==1.13.1",
"tokenizers==0.19.1",
"torch==2.4.0",
"tornado==6.4.1",
"tqdm==4.66.4",
"traitlets==5.14.3",
"transformer-lens==2.3.0",
"transformers==4.43.3",
"triton==3.0.0",
"typeguard==2.13.3",
"typing_extensions==4.12.2",
"tzdata==2024.1",
"urllib3==2.2.2",
"wandb==0.17.5",
"wcwidth==0.2.13",
"xxhash==3.4.1",
"yarl==1.9.4",
],
)