LEAD/requirements.txt at main · HKUSTDial/LEAD · GitHub

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
accelerate==1.1.0
annotated-types==0.7.0
anyio==4.6.2.post1
bitsandbytes==0.42.0
click==8.1.7
cloudpickle==3.1.0
compressed-tensors==0.6.0
conda-pack==0.8.1
deepspeed==0.15.4
diskcache==5.6.3
distro==1.9.0
docker-pycreds==0.4.0
einops==0.8.0
evaluate==0.4.3
exceptiongroup==1.2.2
fastapi==0.115.4
gguf==0.10.0
gitdb==4.0.11
GitPython==3.1.43
h11==0.14.0
hjson==3.1.0
httpcore==1.0.6
httptools==0.6.4
httpx==0.27.2
idna==3.10
importlib_metadata==8.5.0
interegular==0.3.3
Jinja2==3.1.4
jiter==0.7.0
jsonschema==4.23.0
jsonschema-specifications==2024.10.1
lark==1.2.2
llvmlite==0.43.0
lm-format-enforcer==0.10.6
markdown-it-py==3.0.0
MarkupSafe==3.0.2
mdurl==0.1.2
mistral_common==1.4.4
mkl-service==2.4.0
mpmath==1.3.0
msgpack==1.1.0
msgspec==0.18.6
nest-asyncio==1.6.0
networkx==3.4.2
ninja==1.11.1.1
numba==0.60.0
numpy==1.25.2
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.4.127
nvidia-nvtx-cu12==12.1.105
openai==1.54.3
opencv-python-headless==4.10.0.84
outlines==0.0.46
partial-json-parser==0.2.1.1.post4
peft==0.4.0
pillow==10.4.0
platformdirs==4.3.6
prometheus-fastapi-instrumentator==7.0.0
prometheus_client==0.21.0
protobuf==5.28.3
psutil==6.1.0
py-cpuinfo==9.0.0
pyairports==2.1.1
pycountry==24.6.1
pydantic==2.9.2
pydantic_core==2.23.4
Pygments==2.19.1
python-dotenv==1.0.1
pyzmq==26.2.0
ray==2.38.0
referencing==0.35.1
regex==2024.11.6
rich==14.0.0
rpds-py==0.21.0
safetensors==0.4.5
scipy==1.14.1
sentry-sdk==2.18.0
setproctitle==1.3.3
smmap==5.0.1
sniffio==1.3.1
starlette==0.41.2
sympy==1.13.1
tiktoken==0.7.0
tokenizers==0.20.3
torch==2.4.0
torchvision==0.19.0
transformers==4.46.2
triton==3.0.0
uvicorn==0.32.0
uvloop==0.21.0
vllm==0.6.3.post1
wandb==0.18.6
watchfiles==0.24.0
websockets==14.0
xformers==0.0.27.post2
zipp==3.21.0