forked from h2oai/h2o-llmstudio
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Pipfile
65 lines (61 loc) · 1.43 KB
/
Pipfile
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
[[source]]
name = "pypi"
url = "https://pypi.org/simple"
verify_ssl = true
[[source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu117"
verify_ssl = false
[requires]
python_version = "3.10"
[packages]
torch = {index = "pytorch", version = "==2.0.0+cu117"}
torchvision = {index = "pytorch", version = "==0.15.1+cu117"}
torchaudio = {index = "pytorch", version = "==2.0.1+cu117"}
tqdm = "==4.64.0"
transformers = "==4.28.1"
numpy = "==1.23.2"
pandas = "==1.4.2"
scikit-learn = "==1.0.2"
boto3 = "==1.20.24"
SQLAlchemy = "==1.4.36"
dill = "==0.3.4"
pyarrow = "==8.0.0"
kaggle = "==1.5.12"
coolname = "==1.1.0"
bokeh = "==2.4.2"
beautifulsoup4 = "==4.11.1"
sqlitedict = "==1.7.0"
sentencepiece = "==0.1.96"
sacrebleu = "==2.0.0"
toml = "==0.10.2"
pyyaml = "==6.0"
protobuf = "==3.20.3"
fastparquet = "==0.8.1"
gputil = "==1.4.0"
huggingface-hub = "==0.13.4"
starlette= "==0.25.0"
peft = {ref = "main", git = "https://github.com/huggingface/peft"}
bitsandbytes = "==0.38.1"
accelerate = "==0.18.0"
openai = "==0.27.4"
einops = "==0.6.1"
datasets = "==2.11.0"
neptune = "==1.1.1"
Jinja2 = "==3.1.2"
tenacity = "*"
[dev-packages]
pytest = "==7.1.3"
black = "==23.3.0"
coverage = "==7.2.4"
flake8-black = "==0.3.6"
flake8-isort = "==6.0.0"
flake8 = "==6.0.0"
isort = "==5.12.0"
pytest-cov = "==4.0.0"
wheel = "==0.40.0"
pytest-dependency = "==0.5.1"
types-requests = "==2.28.11.17"
types-toml = "==0.10.8.6"
types-pyyaml = "==6.0.12.9"
mypy = "==1.2.0"