generated from VectorInstitute/bootcamp_template
-
Notifications
You must be signed in to change notification settings - Fork 1
/
pyproject.toml
56 lines (51 loc) · 998 Bytes
/
pyproject.toml
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
[tool.poetry]
name = "finetuning-and-alignment"
version = "0.1.0"
description = "Fine-Tuning and Alignment Bootcampt Repository."
authors = ["Your Name <[email protected]>"]
readme = "README.md"
[tool.poetry.dependencies]
python = "^3.10"
pre-commit = "^3.7.0"
datasets = "^2.19.1"
bitsandbytes = "^0.43.1"
transformers = "^4.41.1"
evaluate = "^0.4.2"
ipython = "^8.24.0"
lightning = "^2.2.5"
peft = "^0.11.1"
ruff = "^0.4.5"
scipy = "^1.13.1"
scikit-learn = "^1.5.0"
trl = "^0.8.6"
tqdm = "^4.66.4"
torch = "^2.3.0"
sentencepiece = "^0.2.0"
pandas = "^2.2.2"
nltk = "^3.8.1"
ipykernel = "^6.29.4"
ipywidgets = "^8.1.2"
torchmetrics = "^1.4.0.post0"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.ruff]
line-length = 200
lint.select = ["ALL"]
lint.ignore = [
"ANN101",
"ANN001",
"FBT",
"D103",
"D100",
"PTH",
"ANN201",
"TCH002",
"S101",
"N817",
"TCH001",
"E731",
"PLR0913",
"T201",
"INP001"
]