-
Notifications
You must be signed in to change notification settings - Fork 0
/
environment.yml
174 lines (174 loc) · 4.55 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
name: rm_train
channels:
- defaults
dependencies:
- _libgcc_mutex=0.1=main
- _openmp_mutex=5.1=1_gnu
- bzip2=1.0.8=h5eee18b_6
- ca-certificates=2024.9.24=h06a4308_0
- ld_impl_linux-64=2.40=h12ee557_0
- libffi=3.4.4=h6a678d5_1
- libgcc-ng=11.2.0=h1234567_1
- libgfortran-ng=7.5.0=ha8ba4b0_17
- libgfortran4=7.5.0=ha8ba4b0_17
- libgomp=11.2.0=h1234567_1
- libstdcxx-ng=11.2.0=h1234567_1
- libuuid=1.41.5=h5eee18b_0
- mpi=1.0=mpich
- mpi4py=3.1.4=py39hfc96bbd_0
- mpich=3.3.2=hc856adb_0
- ncurses=6.4=h6a678d5_0
- openssl=3.0.15=h5eee18b_0
- pip=24.2=py39h06a4308_0
- python=3.9.20=he870216_1
- readline=8.2=h5eee18b_0
- setuptools=75.1.0=py39h06a4308_0
- sqlite=3.45.3=h5eee18b_0
- tk=8.6.14=h39e8969_0
- wheel=0.44.0=py39h06a4308_0
- xz=5.4.6=h5eee18b_1
- zlib=1.2.13=h5eee18b_1
- pip:
- accelerate==0.34.2
- aiohappyeyeballs==2.4.0
- aiohttp==3.10.6
- aiosignal==1.3.1
- annotated-types==0.7.0
- anyio==4.6.0
- async-timeout==4.0.3
- attrs==24.2.0
- certifi==2024.8.30
- charset-normalizer==3.3.2
- click==8.1.7
- cloudpickle==3.0.0
- datasets==3.0.1
- deepspeed==0.14.4
- dill==0.3.8
- diskcache==5.6.3
- distro==1.9.0
- docker-pycreds==0.4.0
- docopt==0.6.2
- docstring-parser==0.16
- einops==0.8.0
- eval-type-backport==0.2.0
- evaluate==0.4.3
- exceptiongroup==1.2.2
- fastapi==0.115.0
- fastchat==0.1.0
- filelock==3.16.1
- frozenlist==1.4.1
- fsspec==2023.9.2
- gguf==0.9.1
- gitdb==4.0.11
- gitpython==3.1.43
- h11==0.14.0
- hjson==3.1.0
- httpcore==1.0.6
- httptools==0.6.1
- httpx==0.27.2
- huggingface-hub==0.25.1
- idna==3.10
- importlib-metadata==8.5.0
- interegular==0.3.3
- jinja2==3.1.4
- jiter==0.5.0
- jsonlines==4.0.0
- jsonschema==4.23.0
- jsonschema-specifications==2023.12.1
- lark==1.2.2
- llvmlite==0.43.0
- lm-format-enforcer==0.10.6
- markdown-it-py==3.0.0
- markupsafe==2.1.5
- mdurl==0.1.2
- mistral-common==1.4.4
- mosestokenizer==1.0.0
- mpmath==1.3.0
- msgpack==1.1.0
- msgspec==0.18.6
- multidict==6.1.0
- multiprocess==0.70.16
- nest-asyncio==1.6.0
- networkx==3.2.1
- ninja==1.11.1.1
- numba==0.60.0
- 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-ml-py==12.560.30
- nvidia-nccl-cu12==2.20.5
- nvidia-nvjitlink-cu12==12.6.68
- nvidia-nvtx-cu12==12.1.105
- openai==1.51.0
- openfile==0.0.7
- outlines==0.0.46
- packaging==24.1
- pandas==2.2.3
- partial-json-parser==0.2.1.1.post4
- pillow==10.4.0
- platformdirs==4.3.6
- prometheus-client==0.21.0
- prometheus-fastapi-instrumentator==7.0.0
- protobuf==5.28.2
- psutil==6.0.0
- py-cpuinfo==9.0.0
- pyairports==2.1.1
- pyarrow==17.0.0
- pycountry==24.6.1
- pydantic==2.9.2
- pydantic-core==2.23.4
- pyext==0.5
- pygments==2.18.0
- pynvml==11.5.3
- python-dateutil==2.9.0.post0
- python-dotenv==1.0.1
- pytz==2024.2
- pyyaml==6.0.2
- pyzmq==26.2.0
- ray==2.37.0
- referencing==0.35.1
- regex==2024.9.11
- requests==2.32.3
- rich==13.8.1
- rpds-py==0.20.0
- safetensors==0.4.5
- sentencepiece==0.2.0
- sentry-sdk==2.14.0
- setproctitle==1.3.3
- shtab==1.7.1
- six==1.16.0
- smmap==5.0.1
- sniffio==1.3.1
- starlette==0.38.6
- sympy==1.13.3
- tiktoken==0.7.0
- tokenizers==0.20.0
- toolwrapper==2.1.0
- torch==2.4.0
- torchvision==0.19.0
- tqdm==4.66.5
- transformers==4.45.0
- triton==3.0.0
- trl==0.11.1
- typing-extensions==4.12.2
- tyro==0.8.11
- tzdata==2024.2
- urllib3==2.2.3
- uvicorn==0.31.0
- uvloop==0.20.0
- vllm==0.6.1.post1
- vllm-flash-attn==2.6.1
- wandb==0.18.1
- watchfiles==0.24.0
- websockets==13.1
- xformers==0.0.27.post2
- xxhash==3.5.0
- yarl==1.12.1
- zipp==3.20.2