File size: 750 Bytes
3355706
d69ba2b
768d348
bcc78d8
6a4562a
fce40aa
9e3f0cb
c25ba79
bdfe7c9
ce24f5e
c29117a
6a4562a
0de1457
f2a2029
 
80b2ed2
0de1457
105d0b3
6dd2e7d
0a22847
2bb0b78
c29117a
090c24d
312b8d5
 
 
 
 
 
9c31410
548787d
9a5eb39
6a4562a
b3a61e8
3cc67d2
d69ba2b
 
3cc67d2
 
 
 
f243c21
d7057cc
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
--extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/
packaging==23.2
peft==0.7.0
transformers @ git+https://github.com/huggingface/transformers.git@3cefac1d974db5e2825a0cb2b842883a628be7a0
tokenizers==0.15.0
bitsandbytes>=0.41.1
accelerate @ git+https://github.com/huggingface/accelerate.git@0d2280dadc6a93413a5496613b7fdda3a4d2551b
deepspeed
addict
fire
PyYAML>=6.0
datasets>=2.15.0
flash-attn==2.3.3
sentencepiece
wandb
einops
xformers==0.0.22
optimum==1.13.2
hf_transfer
colorama
numba
numpy>=1.24.4
mlflow
# qlora things
bert-score==0.3.13
evaluate==0.4.0
rouge-score==0.1.2
scipy
scikit-learn==1.2.2
pynvml
art
fschat==0.2.34
gradio==3.50.2
tensorboard

mamba-ssm==1.1.1

# remote filesystems
s3fs
gcsfs
# adlfs

trl>=0.7.9