-
Notifications
You must be signed in to change notification settings - Fork 3
/
environment.yaml
156 lines (156 loc) · 3.99 KB
/
environment.yaml
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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
name: molmo
channels:
- defaults
dependencies:
- _libgcc_mutex=0.1=main
- _openmp_mutex=5.1=1_gnu
- bzip2=1.0.8=h5eee18b_5
- ca-certificates=2024.3.11=h06a4308_0
- ld_impl_linux-64=2.38=h1181459_1
- libffi=3.4.4=h6a678d5_0
- libgcc-ng=11.2.0=h1234567_1
- libgomp=11.2.0=h1234567_1
- libstdcxx-ng=11.2.0=h1234567_1
- libuuid=1.41.5=h5eee18b_0
- ncurses=6.4=h6a678d5_0
- openssl=3.0.13=h7f8727e_0
- pip=24.2=py310h06a4308_0
- python=3.10.14=h955ad1f_0
- readline=8.2=h5eee18b_0
- setuptools=68.2.2=py310h06a4308_0
- sqlite=3.41.2=h5eee18b_0
- tk=8.6.12=h1ccaba5_0
- wheel=0.41.2=py310h06a4308_0
- xz=5.4.6=h5eee18b_0
- zlib=1.2.13=h5eee18b_0
- pip:
- accelerate==0.34.0
- aiohttp==3.9.5
- aiosignal==1.3.1
- annotated-types==0.6.0
- appdirs==1.4.4
- asttokens==2.4.1
- async-timeout==4.0.3
- attrs==23.2.0
- bitsandbytes==0.43.1
- brotli==1.0.9
- certifi==2024.2.2
- charset-normalizer==2.0.4
- click==8.1.7
- comm==0.2.2
- datasets==2.19.2
- debugpy==1.6.7
- decorator==5.1.1
- deepspeed==0.14.0
- dill==0.3.8
- docker-pycreds==0.4.0
- docstring-parser==0.16
- einops==0.7.0
- entrypoints==0.4
- exceptiongroup==1.2.0
- executing==2.0.1
- filelock==3.13.1
- frozenlist==1.4.1
- fsspec==2024.3.1
- ftfy==6.2.0
- gitdb==4.0.11
- gitpython==3.1.43
- gmpy2==2.1.2
- hjson==3.1.0
- huggingface-hub==0.23.2
- idna==3.4
- intel-openmp==2024.1.0
- ipykernel==6.29.3
- ipython==8.22.2
- jedi==0.19.1
- jinja2==3.1.3
- jupyter-client==7.3.4
- jupyter-core==5.7.2
- markdown-it-py==3.0.0
- markupsafe==2.1.3
- matplotlib-inline==0.1.6
- mdurl==0.1.2
- mkl==2024.1.0
- mkl-fft==1.3.8
- mkl-random==1.2.4
- mkl-service==2.4.0
- mpmath==1.3.0
- multidict==6.0.5
- multiprocess==0.70.16
- nest-asyncio==1.6.0
- networkx==3.1
- ninja==1.11.1.1
- 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==8.9.2.26
- 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-ml-py==12.560.30
- nvidia-nccl-cu12==2.20.5
- nvidia-nvjitlink-cu12==12.4.127
- nvidia-nvtx-cu12==12.1.105
- packaging==24.0
- pandas==2.2.2
- parso==0.8.4
- peft==0.10.0
- pexpect==4.9.0
- pickleshare==0.7.5
- pillow==10.2.0
- platformdirs==4.2.0
- prompt-toolkit==3.0.42
- protobuf==4.25.3
- psutil==5.9.0
- ptyprocess==0.7.0
- pure-eval==0.2.2
- py-cpuinfo==9.0.0
- pyarrow==16.1.0
- pyarrow-hotfix==0.6
- pydantic==2.6.4
- pydantic-core==2.16.3
- pygments==2.17.2
- pynvml==11.5.0
- pysocks==1.7.1
- python-dateutil==2.9.0
- pytz==2024.1
- pyyaml==6.0.1
- pyzmq==25.1.2
- regex==2023.12.25
- requests==2.32.3
- rich==13.7.1
- safetensors==0.4.5
- sentencepiece==0.2.0
- sentry-sdk==1.45.0
- setproctitle==1.3.3
- shtab==1.7.1
- six==1.16.0
- smmap==5.0.1
- stack-data==0.6.2
- sympy==1.12
- tbb==2021.12.0
- tensorboardx==2.6.2.2
- tokenizers==0.20.0
- torch==2.3.1
- torchaudio==2.3.1
- torchvision==0.18.1
- tornado==6.1
- transformers==4.45.1
- tqdm==4.66.4
- traitlets==5.14.2
- triton==2.3.1
- trl==0.8.6
- typing-extensions==4.9.0
- tyro==0.8.4
- tzdata==2024.1
- ujson==5.10.0
- urllib3==2.1.0
- wandb==0.16.6
- wcwidth==0.2.13
- xxhash==3.4.1
- yarl==1.9.4
- lomo-optim==0.1.1
- decord==0.6.0