-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrequirements.txt
104 lines (104 loc) · 2.08 KB
/
requirements.txt
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
accelerate==0.28.0
aiohttp==3.9.3
aiosignal==1.3.1
annotated-types==0.6.0
appdirs==1.4.4
async-timeout==4.0.3
attrs==23.2.0
certifi==2024.2.2
charset-normalizer==3.3.2
click==8.1.7
cmake==3.25.0
colorama==0.4.6
datasets==2.18.0
deepspeed==0.14.0
dill==0.3.8
docker-pycreds==0.4.0
docstring_parser==0.16
einops==0.7.0
entmax==1.3
evaluate==0.4.1
filelock==3.13.1
-e git+https://github.com/Dao-AILab/flash-attention.git@3566596ad867ee415dd3c12616dd50c610176f6c#egg=flash_attn
frozenlist==1.4.1
fsspec==2024.2.0
gitdb==4.0.11
GitPython==3.1.42
hjson==3.1.0
huggingface-hub==0.21.4
idna==3.6
Jinja2==3.1.3
jsonargparse==3.13.1
lightning-utilities==0.10.1
lit==15.0.7
lxml==5.1.0
markdown-it-py==3.0.0
MarkupSafe==2.1.5
mdurl==0.1.2
mpmath==1.3.0
multidict==6.0.5
multiprocess==0.70.16
networkx==3.2.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-nccl-cu12==2.19.3
nvidia-nvjitlink-cu12==12.4.99
nvidia-nvtx-cu12==12.1.105
packaging==24.0
pandas==2.2.1
peft==0.9.0
pillow==10.2.0
portalocker==2.8.2
protobuf==4.25.3
psutil==5.9.8
py-cpuinfo==9.0.0
pyarrow==15.0.1
pyarrow-hotfix==0.6
pydantic==2.6.4
pydantic_core==2.16.3
Pygments==2.17.2
pynvml==11.5.0
python-dateutil==2.9.0.post0
pytorch-lightning==2.2.1
pytz==2024.1
PyYAML==6.0.1
regex==2023.12.25
requests==2.31.0
responses==0.18.0
rich==13.7.1
sacrebleu==2.4.1
safetensors==0.4.2
scipy==1.12.0
sentencepiece==0.1.99
sentry-sdk==1.42.0
setproctitle==1.3.3
shtab==1.7.1
six==1.16.0
smmap==5.0.1
sympy==1.12
tabulate==0.9.0
tokenizers==0.15.2
torch==2.0.1+cu117
torchmetrics==0.10.3
torchvision==0.15.2+cu117
tqdm==4.66.2
transformers==4.38.2
triton==2.0.0
trl==0.7.11
typing_extensions==4.10.0
tyro==0.7.3
tzdata==2024.1
unbabel-comet==2.2.2
urllib3==2.2.1
wandb==0.16.4
xxhash==3.4.1
yarl==1.9.4