-
Notifications
You must be signed in to change notification settings - Fork 0
/
environment.yml
329 lines (329 loc) · 10.4 KB
/
environment.yml
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
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
name: llamarec
channels:
- pytorch
- nvidia
- conda-forge
dependencies:
- _libgcc_mutex=0.1=conda_forge
- _openmp_mutex=4.5=2_kmp_llvm
- _sysroot_linux-64_curr_repodata_hack=3=h69a702a_14
- absl-py=2.1.0=pyhd8ed1ab_0
- accelerate=0.31.0=pyhd8ed1ab_0
- aiohttp=3.9.5=py311h459d7ec_0
- aiosignal=1.3.1=pyhd8ed1ab_0
- annotated-types=0.7.0=pyhd8ed1ab_0
- astroid=3.2.2=py311h38be061_0
- attrs=23.2.0=pyh71513ae_0
- aws-c-auth=0.7.22=heee8711_7
- aws-c-cal=0.7.0=h816f305_0
- aws-c-common=0.9.23=h4ab18f5_0
- aws-c-compression=0.2.18=he027950_7
- aws-c-event-stream=0.4.2=hb72ac1a_14
- aws-c-http=0.8.2=had8cc17_4
- aws-c-io=0.14.9=h37d6bf3_5
- aws-c-mqtt=0.10.4=hb0abfc5_7
- aws-c-s3=0.6.0=h1f67ec3_0
- aws-c-sdkutils=0.1.16=he027950_3
- aws-checksums=0.1.18=he027950_7
- aws-crt-cpp=0.27.1=h93f15ff_1
- aws-sdk-cpp=1.11.329=h70d5bd1_7
- azure-core-cpp=1.12.0=h830ed8b_0
- azure-identity-cpp=1.8.0=hdb0d106_1
- azure-storage-blobs-cpp=12.11.0=ha67cba7_1
- azure-storage-common-cpp=12.6.0=he3f277c_1
- azure-storage-files-datalake-cpp=12.10.0=h29b5301_1
- binutils_impl_linux-64=2.40=ha1999f0_7
- binutils_linux-64=2.40=hb3c18ed_9
- black=24.4.2=py311h38be061_0
- brotli-python=1.1.0=py311hb755f60_1
- bzip2=1.0.8=hd590300_5
- c-ares=1.28.1=hd590300_0
- ca-certificates=2024.7.4=hbcca054_0
- certifi=2024.6.2=pyhd8ed1ab_0
- cffi=1.16.0=py311hb3a22ac_0
- charset-normalizer=3.3.2=pyhd8ed1ab_0
- click=8.1.7=unix_pyh707e725_0
- colorama=0.4.6=pyhd8ed1ab_0
- cuda-cccl=12.5.39=0
- cuda-cccl_linux-64=12.5.39=0
- cuda-command-line-tools=12.1.1=0
- cuda-compiler=12.5.0=0
- cuda-crt-dev_linux-64=12.5.40=0
- cuda-crt-tools=12.5.40=0
- cuda-cudart=12.1.105=0
- cuda-cudart-dev=12.1.105=0
- cuda-cudart-dev_linux-64=12.5.39=0
- cuda-cudart-static=12.5.39=0
- cuda-cudart-static_linux-64=12.5.39=0
- cuda-cudart_linux-64=12.5.39=0
- cuda-cuobjdump=12.5.39=0
- cuda-cupti=12.1.105=0
- cuda-cuxxfilt=12.5.39=0
- cuda-documentation=12.4.127=0
- cuda-driver-dev=12.5.39=0
- cuda-driver-dev_linux-64=12.5.39=0
- cuda-gdb=12.5.39=0
- cuda-libraries=12.1.0=0
- cuda-libraries-dev=12.5.0=0
- cuda-libraries-static=12.5.0=0
- cuda-nsight=12.5.39=0
- cuda-nvcc=12.5.40=0
- cuda-nvcc-dev_linux-64=12.5.40=0
- cuda-nvcc-impl=12.5.40=0
- cuda-nvcc-tools=12.5.40=0
- cuda-nvcc_linux-64=12.5.40=0
- cuda-nvdisasm=12.5.39=0
- cuda-nvml-dev=12.5.39=0
- cuda-nvprof=12.5.39=0
- cuda-nvprune=12.5.39=0
- cuda-nvrtc=12.1.105=0
- cuda-nvrtc-dev=12.1.105=0
- cuda-nvrtc-static=12.5.40=0
- cuda-nvtx=12.1.105=0
- cuda-nvvm-dev_linux-64=12.5.40=0
- cuda-nvvm-impl=12.5.40=0
- cuda-nvvm-tools=12.5.40=0
- cuda-nvvp=12.5.39=0
- cuda-opencl=12.5.39=0
- cuda-opencl-dev=12.5.39=0
- cuda-profiler-api=12.5.39=0
- cuda-runtime=12.1.0=0
- cuda-sanitizer-api=12.5.39=0
- cuda-toolkit=12.1.0=0
- cuda-tools=12.1.1=0
- cuda-version=12.5=3
- cuda-visual-tools=12.5.0=0
- cudnn=8.9.7.29=h092f7fd_3
- datasets=2.20.0=pyhd8ed1ab_0
- dill=0.3.8=pyhd8ed1ab_0
- docker-pycreds=0.4.0=py_0
- docstring_parser=0.16=pyhd8ed1ab_0
- einops=0.8.0=pyhd8ed1ab_0
- filelock=3.15.4=pyhd8ed1ab_0
- flash-attn=2.5.8=py311h0e90392_1
- freetype=2.12.1=h267a509_2
- frozendict=2.4.4=py311h331c9d8_0
- frozenlist=1.4.1=py311h459d7ec_0
- fsspec=2024.5.0=pyhff2d567_0
- gcc_impl_linux-64=12.3.0=h58ffeeb_13
- gcc_linux-64=12.3.0=h9528a6a_9
- gds-tools=1.10.0.4=0
- gflags=2.2.2=he1b5a44_1004
- glog=0.7.1=hbabe93e_0
- gmp=6.3.0=hac33072_2
- gmpy2=2.1.5=py311hc4f1f91_1
- grpcio=1.62.2=py311ha6695c7_0
- gxx_impl_linux-64=12.3.0=h2a574ab_13
- gxx_linux-64=12.3.0=ha28b414_9
- h2=4.1.0=pyhd8ed1ab_0
- hjson-py=3.1.0=pyhd8ed1ab_0
- hpack=4.0.0=pyh9f0ad1d_0
- huggingface_hub=0.23.4=pyhd8ed1ab_0
- hyperframe=6.0.1=pyhd8ed1ab_0
- icu=73.2=h59595ed_0
- idna=3.7=pyhd8ed1ab_0
- importlib-metadata=8.0.0=pyha770c72_0
- isort=5.13.2=pyhd8ed1ab_0
- jinja2=3.1.4=pyhd8ed1ab_0
- kernel-headers_linux-64=3.10.0=h4a8ded7_14
- keyutils=1.6.1=h166bdaf_0
- krb5=1.21.3=h659f571_0
- lcms2=2.16=hb7c19ff_0
- ld_impl_linux-64=2.40=hf3520f5_7
- lerc=4.0.0=h27087fc_0
- libabseil=20240116.2=cxx17_h59595ed_0
- libarrow=16.1.0=h040edfd_12_cpu
- libarrow-acero=16.1.0=he02047a_12_cpu
- libarrow-dataset=16.1.0=he02047a_12_cpu
- libarrow-substrait=16.1.0=hc9a23c6_12_cpu
- libblas=3.9.0=22_linux64_openblas
- libbrotlicommon=1.1.0=hd590300_1
- libbrotlidec=1.1.0=hd590300_1
- libbrotlienc=1.1.0=hd590300_1
- libcblas=3.9.0=22_linux64_openblas
- libcrc32c=1.1.2=h9c3ff4c_0
- libcublas=12.1.0.26=0
- libcublas-dev=12.1.0.26=0
- libcublas-static=12.5.2.13=0
- libcufft=11.0.2.4=0
- libcufft-dev=11.0.2.4=0
- libcufft-static=11.2.3.18=0
- libcufile=1.10.0.4=0
- libcufile-dev=1.10.0.4=0
- libcufile-static=1.10.0.4=0
- libcurand=10.3.6.39=0
- libcurand-dev=10.3.6.39=0
- libcurand-static=10.3.6.39=0
- libcurl=8.8.0=hca28451_1
- libcusolver=11.4.4.55=0
- libcusolver-dev=11.4.4.55=0
- libcusolver-static=11.6.2.40=0
- libcusparse=12.0.2.55=0
- libcusparse-dev=12.0.2.55=0
- libcusparse-static=12.4.1.24=0
- libdeflate=1.20=hd590300_0
- libedit=3.1.20191231=he28a2e2_2
- libev=4.33=hd590300_2
- libevent=2.1.12=hf998b51_1
- libexpat=2.6.2=h59595ed_0
- libffi=3.4.2=h7f98852_5
- libgcc-devel_linux-64=12.3.0=h6b66f73_113
- libgcc-ng=14.1.0=h77fa898_0
- libgfortran-ng=14.1.0=h69a702a_0
- libgfortran5=14.1.0=hc5f4f2c_0
- libgomp=14.1.0=h77fa898_0
- libgoogle-cloud=2.26.0=h26d7fe4_0
- libgoogle-cloud-storage=2.26.0=ha262f82_0
- libgrpc=1.62.2=h15f2491_0
- libhwloc=2.11.0=default_h5622ce7_1000
- libiconv=1.17=hd590300_2
- libjpeg-turbo=3.0.0=hd590300_1
- liblapack=3.9.0=22_linux64_openblas
- libmagma=2.7.2=h173bb3b_2
- libmagma_sparse=2.7.2=h173bb3b_3
- libnghttp2=1.58.0=h47da74e_1
- libnpp=12.0.2.50=0
- libnpp-dev=12.0.2.50=0
- libnpp-static=12.3.0.116=0
- libnsl=2.0.1=hd590300_0
- libnvfatbin=12.5.39=0
- libnvfatbin-dev=12.5.39=0
- libnvfatbin-static=12.5.39=0
- libnvjitlink=12.1.105=0
- libnvjitlink-dev=12.1.105=0
- libnvjitlink-static=12.5.40=0
- libnvjpeg=12.1.1.14=0
- libnvjpeg-dev=12.1.1.14=0
- libnvjpeg-static=12.3.2.38=0
- libnvvm-samples=12.1.105=0
- libopenblas=0.3.27=pthreads_h413a1c8_0
- libparquet=16.1.0=h9e5060d_12_cpu
- libpng=1.6.43=h2797004_0
- libprotobuf=4.25.3=h08a7969_0
- libre2-11=2023.09.01=h5a48ba9_2
- libsanitizer=12.3.0=hb8811af_13
- libsentencepiece=0.2.0=he81a138_2
- libsqlite=3.46.0=hde9e2c9_0
- libssh2=1.11.0=h0841786_0
- libstdcxx-devel_linux-64=12.3.0=h6b66f73_113
- libstdcxx-ng=14.1.0=hc0a3c3a_0
- libthrift=0.19.0=hb90f79a_1
- libtiff=4.6.0=h1dd3fc0_3
- libtorch=2.3.1=cuda120_h2b0da52_300
- libutf8proc=2.8.0=h166bdaf_0
- libuuid=2.38.1=h0b41bf4_0
- libuv=1.48.0=hd590300_0
- libwebp-base=1.4.0=hd590300_0
- libxcb=1.16=hd590300_0
- libxcrypt=4.4.36=hd590300_1
- libxml2=2.12.7=hc051c1a_1
- libzlib=1.3.1=h4ab18f5_1
- lightning=2.3.1=pyhd8ed1ab_0
- lightning-utilities=0.11.3.post0=pyhd8ed1ab_0
- llvm-openmp=18.1.8=hf5423f3_0
- lz4-c=1.9.4=hcb278e6_0
- markdown=3.6=pyhd8ed1ab_0
- markdown-it-py=3.0.0=pyhd8ed1ab_0
- markupsafe=2.1.5=py311h459d7ec_0
- mccabe=0.7.0=pyhd8ed1ab_0
- mdurl=0.1.2=pyhd8ed1ab_0
- mkl=2023.2.0=h84fe81f_50496
- mpc=1.3.1=hfe3b2da_0
- mpfr=4.2.1=h9458935_1
- mpmath=1.3.0=pyhd8ed1ab_0
- multidict=6.0.5=py311h459d7ec_0
- multiprocess=0.70.16=py311h459d7ec_0
- mypy_extensions=1.0.0=pyha770c72_0
- nccl=2.22.3.1=hbc370b7_0
- ncurses=6.5=h59595ed_0
- networkx=3.3=pyhd8ed1ab_1
- nsight-compute=2024.1.1.4=0
- numpy=1.26.4=py311h64a7726_0
- openjpeg=2.5.2=h488ebb8_0
- openssl=3.3.1=h4ab18f5_1
- orc=2.0.1=h17fec99_1
- packaging=24.1=pyhd8ed1ab_0
- pandas=2.2.2=py311h14de704_1
- pathspec=0.12.1=pyhd8ed1ab_0
- peft=0.11.1=pyhd8ed1ab_0
- pillow=10.4.0=py311h82a398c_0
- pip=24.0=pyhd8ed1ab_0
- platformdirs=4.2.2=pyhd8ed1ab_0
- protobuf=4.25.3=py311h7b78aeb_0
- psutil=6.0.0=py311h331c9d8_0
- pthread-stubs=0.4=h36c2ea0_1001
- py-cpuinfo=9.0.0=pyhd8ed1ab_0
- pyarrow=16.1.0=py311hbd00459_4
- pyarrow-core=16.1.0=py311h8c3dac4_4_cpu
- pyarrow-hotfix=0.6=pyhd8ed1ab_0
- pycparser=2.22=pyhd8ed1ab_0
- pydantic=2.8.0=pyhd8ed1ab_0
- pydantic-core=2.20.0=py311h5ecf98a_0
- pygments=2.18.0=pyhd8ed1ab_0
- pylint=3.2.5=pyhd8ed1ab_0
- pyre-extensions=0.0.29=pyhd8ed1ab_0
- pysocks=1.7.1=pyha2e5f31_6
- python=3.11.9=hb806964_0_cpython
- python-dateutil=2.9.0=pyhd8ed1ab_0
- python-tzdata=2024.1=pyhd8ed1ab_0
- python-xxhash=3.4.1=py311h459d7ec_0
- python_abi=3.11=4_cp311
- pytorch=2.3.1=cuda120_py311hf6aebf0_300
- pytorch-cuda=12.1=ha16c6d3_5
- pytorch-lightning=2.3.0=pyhd8ed1ab_0
- pytz=2024.1=pyhd8ed1ab_0
- pyyaml=6.0.1=py311h459d7ec_1
- re2=2023.09.01=h7f4b329_2
- readline=8.2=h8228510_1
- regex=2024.5.15=py311h331c9d8_0
- requests=2.32.3=pyhd8ed1ab_0
- rich=13.7.1=pyhd8ed1ab_0
- s2n=1.4.17=he19d79f_0
- safetensors=0.4.3=py311h46250e7_0
- scipy=1.14.0=py311h517d4fd_0
- sentencepiece=0.2.0=h38be061_2
- sentencepiece-python=0.2.0=py311h7fa642f_2
- sentencepiece-spm=0.2.0=he81a138_2
- sentry-sdk=2.7.1=pyhd8ed1ab_0
- setuptools=70.1.1=pyhd8ed1ab_0
- shtab=1.7.1=pyhd8ed1ab_0
- six=1.16.0=pyh6c4a22f_0
- sleef=3.5.1=h9b69904_2
- snappy=1.2.1=ha2e4443_0
- sympy=1.12.1=pypyh2585a3b_103
- sysroot_linux-64=2.17=h4a8ded7_14
- tbb=2021.12.0=h434a139_2
- tensorboard=2.17.0=pyhd8ed1ab_0
- tensorboard-data-server=0.7.0=py311h63ff55d_1
- tk=8.6.13=noxft_h4845f30_101
- tokenizers=0.19.1=py311h6640629_0
- tomli=2.0.1=pyhd8ed1ab_0
- tomlkit=0.12.5=pyha770c72_0
- torchmetrics=1.4.0.post0=pyhd8ed1ab_0
- tqdm=4.66.4=pyhd8ed1ab_0
- transformers=4.42.3=pyhd8ed1ab_0
- trl=0.9.4=pyhd8ed1ab_0
- typing-extensions=4.12.2=hd8ed1ab_0
- typing_extensions=4.12.2=pyha770c72_0
- typing_inspect=0.9.0=pyhd8ed1ab_0
- tyro=0.8.5=pyhd8ed1ab_0
- tzdata=2024a=h0c530f3_0
- urllib3=2.2.2=pyhd8ed1ab_1
- werkzeug=3.0.3=pyhd8ed1ab_0
- wheel=0.43.0=pyhd8ed1ab_1
- whoosh=2.7.4=py311h38be061_8
- xformers=0.0.26.post1=cuda120py311h7da0735_0
- xorg-libxau=1.0.11=hd590300_0
- xorg-libxdmcp=1.1.3=h7f98852_0
- xxhash=0.8.2=hd590300_0
- xz=5.2.6=h166bdaf_0
- yacs=0.1.8=pyhd8ed1ab_0
- yaml=0.2.5=h7f98852_2
- yarl=1.9.4=py311h459d7ec_0
- zipp=3.19.2=pyhd8ed1ab_0
- zstandard=0.22.0=py311hb6f056b_1
- zstd=1.5.6=ha6fb4c9_0
- pip:
- bitsandbytes==0.43.1
- triton==2.3.1
- wandb==0.17.4