GERM/requirements.txt at main · MAGICS-LAB/GERM · 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
accelerate
attributedict
attrs
auto_gptq
bert-score==0.3.13
bioc==2.1
biopython==1.83
bitsandbytes==0.42.0
cmake==3.27.7
compressed-tensors==0.6.0
datasets==2.21.0
deepdiff==6.7.1
deepspeed==0.15.1
Deprecated==1.2.14
dill==0.3.6
distlib==0.3.7
easydict==1.13
einops==0.8.0
evaluate==0.4.2
fastapi==0.112.4
flair==0.14.0
gdown==5.2.0
huggingface-hub==0.24.6
language_tool_python==2.8.1
lmppl==0.3.2
matplotlib==3.7.5
mistral_common==1.4.4
multidict==6.0.4
multiprocess==0.70.14
nltk==3.8.1
num2words==0.5.13
numpy
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-cu11==10.9.0.58
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.77
nvidia-nvtx-cu12==12.1.105
-e git+https://github.com/OpenGVLab/OmniQuant.git@8eca0ce9ae222e344e5ce0d8897ed9e2b3404b6e#egg=omniquant
opencv-python-headless==4.10.0.84
packaging==23.2
pandas==2.0.3
partial-json-parser==0.2.1.1.post4
peft==0.12.0
pillow
pluggy==1.3.0
pyarrow==17.0.0
pyparsing==3.1.4
PySocks
pytorch_revgrad==0.2.0
pytz==2023.3
PyYAML
pyzmq==26.2.0
ray==2.10.0
regex
rich==13.8.0
rootpath==0.1.1
rouge==1.0.1
rouge-score==0.1.2
sacrebleu==1.5.0
safetensors==0.4.4
scikit-learn==1.3.2
scipy==1.10.1
seaborn==0.13.2
sentencepiece==0.2.0
six==1.16.0
smoothquant==0.0.0
sqlitedict==2.1.0
starlette==0.38.6
sympy==1.12
tensorboard==2.14.0
tensorboard-data-server==0.7.2
tensorboardX==2.6.2.2
textattack==0.3.10
threadpoolctl==3.2.0
tiktoken==0.7.0
timm==0.4.5
tokenizers==0.20.1
toml==0.10.2
tomli==2.0.1
torch==2.4.0+cu121
torch_int==0.0.0
torchaudio==2.4.0+cu121
torchvision==0.19.0
tox==4.11.4
tqdm==4.66.5
tqdm-multiprocess==0.0.11
transformer-smaller-training-vocab==0.4.0
transformer_engine==0.0.0
transformers==4.45.2
typing_extensions
wandb
xformers