forked from h2oai/h2o-llmstudio
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Pipfile
71 lines (67 loc) · 1.58 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
71
[[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"
neptune-client = "==0.16.1"
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.37.2"
accelerate = "==0.18.0"
openai = "==0.27.4"
einops = "==0.6.1"
datasets = "==2.11.0"
[dev-packages]
pytest = "==7.1.2"
black = "==22.3.0"
coverage = "==6.3.2"
flake8-black = "==0.3.2"
flake8-isort = "==4.1.1"
flake8 = "==4.0.1"
isort = "==5.10.1"
pytest-cov = "==3.0.0"
wheel = "==0.38.1"
pytest-dependency = "==0.5.1"
types-requests = "==2.27.25"
types-toml = "==0.10.7"
types-pyyaml = "==6.0.7"
mypy = "==0.950"
pytest-xprocess = "==0.18.1"
mkdocs-material = "==8.2.14"
mkdocs-macros-plugin = "==0.7.0"
mike = "==1.1.2"
termcolor = "==1.1.0"
pip = "==22.0.4"
pipenv = "==2022.5.2"
setuptools = "62.1.0"