-
Notifications
You must be signed in to change notification settings - Fork 1
/
environment_dpo.yaml
114 lines (114 loc) · 2.93 KB
/
environment_dpo.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
name: dpo
channels:
- pytorch
- nvidia
- defaults
dependencies:
- _libgcc_mutex=0.1=main
- _openmp_mutex=5.1=1_gnu
- blas=1.0=mkl
- ca-certificates=2023.08.22=h06a4308_0
- cuda-cudart=11.8.89=0
- cuda-cupti=11.8.87=0
- cuda-libraries=11.8.0=0
- cuda-nvrtc=11.8.89=0
- cuda-nvtx=11.8.86=0
- cuda-runtime=11.8.0=0
- filelock=3.13.1=py39h06a4308_0
- gmp=6.2.1=h295c915_3
- gmpy2=2.1.2=py39heeb90bb_0
- intel-openmp=2023.1.0=hdb19cb5_46306
- jinja2=3.1.2=py39h06a4308_0
- ld_impl_linux-64=2.38=h1181459_1
- libcublas=11.11.3.6=0
- libcufft=10.9.0.58=0
- libcufile=1.8.1.2=0
- libcurand=10.3.4.101=0
- libcusolver=11.4.1.48=0
- libcusparse=11.7.5.86=0
- libffi=3.4.4=h6a678d5_0
- libgcc-ng=11.2.0=h1234567_1
- libgomp=11.2.0=h1234567_1
- libnpp=11.8.0.86=0
- libnvjpeg=11.9.0.86=0
- libstdcxx-ng=11.2.0=h1234567_1
- llvm-openmp=14.0.6=h9e868ea_0
- markupsafe=2.1.1=py39h7f8727e_0
- mkl=2023.1.0=h213fc3f_46344
- mpc=1.1.0=h10f8cd9_1
- mpfr=4.0.2=hb69a4c5_1
- mpmath=1.3.0=py39h06a4308_0
- ncurses=6.4=h6a678d5_0
- networkx=3.1=py39h06a4308_0
- openssl=3.0.12=h7f8727e_0
- pip=23.3.1=py39h06a4308_0
- python=3.9.18=h955ad1f_0
- pytorch=2.1.1=py3.9_cuda11.8_cudnn8.7.0_0
- pytorch-cuda=11.8=h7e8668a_5
- pytorch-mutex=1.0=cuda
- pyyaml=6.0.1=py39h5eee18b_0
- readline=8.2=h5eee18b_0
- setuptools=68.0.0=py39h06a4308_0
- sqlite=3.41.2=h5eee18b_0
- sympy=1.11.1=py39h06a4308_0
- tbb=2021.8.0=hdb19cb5_0
- tk=8.6.12=h1ccaba5_0
- torchtriton=2.1.0=py39
- typing_extensions=4.7.1=py39h06a4308_0
- wheel=0.41.2=py39h06a4308_0
- xz=5.4.2=h5eee18b_0
- yaml=0.2.5=h7b6447c_0
- zlib=1.2.13=h5eee18b_0
- pip:
- accelerate==0.27.2
- aiohttp==3.9.1
- aiosignal==1.3.1
- appdirs==1.4.4
- async-timeout==4.0.3
- attrs==23.1.0
- certifi==2023.11.17
- charset-normalizer==3.3.2
- click==8.1.7
- datasets==2.15.0
- dill==0.3.7
- docker-pycreds==0.4.0
- docstring-parser==0.15
- frozenlist==1.4.0
- fsspec==2023.10.0
- gitdb==4.0.11
- gitpython==3.1.40
- huggingface-hub==0.19.4
- idna==3.6
- markdown-it-py==3.0.0
- mdurl==0.1.2
- multidict==6.0.4
- multiprocess==0.70.15
- numpy==1.26.2
- packaging==23.2
- pandas==2.1.3
- protobuf==4.25.1
- psutil==5.9.6
- pyarrow==14.0.1
- pyarrow-hotfix==0.6
- pygments==2.17.2
- python-dateutil==2.8.2
- pytz==2023.3.post1
- regex==2023.10.3
- requests==2.31.0
- rich==13.7.0
- safetensors==0.4.1
- sentry-sdk==1.37.1
- setproctitle==1.3.3
- shtab==1.6.4
- six==1.16.0
- smmap==5.0.1
- tokenizers==0.15.0
- tqdm==4.66.1
- transformers==4.38.2
- trl==0.7.11
- tyro==0.5.18
- tzdata==2023.3
- urllib3==2.1.0
- wandb==0.16.0
- xxhash==3.4.1
- yarl==1.9.3