forked from h2oai/h2o-llmstudio
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Pipfile
70 lines (66 loc) · 1.59 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
66
67
68
69
70
[[source]]
name = "pypi"
url = "https://pypi.org/simple"
verify_ssl = true
[[source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu118"
verify_ssl = false
[requires]
python_version = "3.10"
[packages]
torch = {index = "pytorch", version = "==2.1.2+cu118"}
tqdm = ">=4.65.0, <5.0.0"
transformers = "==4.36.1"
numpy = ">=1.23.2, <2.0.0"
pandas = ">=2.1.0, <3.0.0"
scikit-learn = ">=1.0.2, <2.0.0"
boto3 = ">=1.20.24, <2.0.0"
SQLAlchemy = ">=1.4.36, <2.0.0"
dill = ">=0.3.4, <0.4.0"
pyarrow = ">=14.0.1"
kaggle = ">=1.5.12, <2.0.0"
coolname = ">=2.2.0, <3.0.0"
bokeh = ">=3.2.1, <4.0.0"
beautifulsoup4 = ">=4.11.1, <5.0.0"
sqlitedict = "==1.7.0"
sentencepiece = ">=0.1.96, <0.2.0"
sacrebleu = "==2.0.0"
toml = ">=0.10.2, <0.11.0"
pyyaml = ">=6.0.0, <7.0.0"
protobuf = "==3.20.3"
fastparquet = ">=2023.7.0"
gputil = ">=1.4.0, <2.0.0"
huggingface-hub = "==0.19.3"
starlette= ">=0.31.0, <0.32.0"
bitsandbytes = "==0.41.1"
accelerate = "==0.23.0"
openai = ">=0.28.0, <0.29.0"
einops = "==0.6.1"
datasets = ">=2.11.0, <3.0.0"
neptune = "==1.6.3"
Jinja2 = ">=3.1.2, <4.0.0"
tenacity = ">=8.2.2, <9.0.0"
h2o-wave = "==0.26.2"
tiktoken = "==0.5.1"
hf-transfer = "==0.1.3"
peft = "==0.5.0"
azure-storage-file-datalake = ">=12.12.0"
deepspeed = "==0.11.1"
keyring = "==24.2.0"
[dev-packages]
black = "==23.7.0"
coverage = "==7.2.7"
flake8 = "==6.1.0"
flake8-black = "==0.3.6"
flake8-isort = "==6.0.0"
isort = "==5.12.0"
mypy = "==1.4.1"
pytest = "==7.4.0"
pytest-cov = "==4.1.0"
pytest-dependency = "==0.5.1"
pytest-html = "3.2.0"
types-pyyaml = ">=6.0"
types-requests = ">=2.31"
types-toml = ">=0.10"
wheel = "==0.41.0"