-
Notifications
You must be signed in to change notification settings - Fork 1
/
pyproject.toml
74 lines (64 loc) · 1.63 KB
/
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
[tool.poetry]
name = "docugami_dfm_benchmarks"
version = "0.0.3"
description = "Benchmarks for Docugami's Document Foundation Model."
authors = ["Docugami Inc. <contact@docugami.com>"]
packages = [{ include = "docugami_dfm_benchmarks" }]
[tool.poetry.dependencies]
python = ">=3.9,<3.11"
torch = "^2.2.1"
transformers = "^4.38.2"
sentence-transformers = "^2.5.1"
pyyaml = "^6.0.1"
importlib-metadata = "^7.0.2"
docugami-langchain = {git = "https://github.com/docugami/docugami-langchain.git", rev = "tjaffri/typing"}
langchain-fireworks = "^0.1.1"
langchain-community = "^0.0.30"
faiss-cpu = "^1.8.0"
accelerate = "^0.28.0"
bitsandbytes = "^0.43.0"
[tool.poetry.group.dev.dependencies]
ipykernel = "^6.29.3"
tabulate = "^0.9.0"
tqdm = "^4.66.2"
notebook = "^7.1.1"
ipywidgets = "^8.1.2"
setuptools = "^69.2.0"
ruff = ">=0.3.0"
mypy = "^1.9.0"
codespell = "^2.2.6"
pytest = "^8.1.1"
typer = "^0.9.0"
types-tabulate = "^0.9.0.20240106"
types-tqdm = "^4.66.0.20240106"
[tool.poetry.group.training]
optional = true
[tool.poetry.group.training.dependencies]
accelerate = "^0.28.0"
bitsandbytes = "^0.43.0"
loralib = "^0.1.2"
datasets = ">=2.18.0"
evaluate = ">=0.4.1"
dgml-utils = ">=0.3.2"
peft = ">=0.10.0"
[tool.ruff]
lint.select = [
"E", # pycodestyle
"F", # pyflakes
"I", # isort
]
line-length = 200
[tool.mypy]
disallow_untyped_defs = "True"
disable_error_code = "typeddict-item"
[[tool.mypy.overrides]]
module = [
"sentence_transformers",
"transformers"
]
ignore_missing_imports = true
[tool.poetry.scripts]
benchmark = "docugami_dfm_benchmarks.cli:app"
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"