-
Notifications
You must be signed in to change notification settings - Fork 3
/
pyproject.toml
77 lines (71 loc) · 2.58 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
75
76
77
[tool.poetry]
name = "docmae"
version = "0.1"
description = "Unofficial implementation of DocMAE paper"
authors = ["David Komorowicz <dawars00@gmail.com>"]
packages = [
{ include = "docmae" } # dir name containing code if differs from name
]
[tool.poetry.dependencies]
python = "~3.10"
opencv-python-headless = "^4.5"
Pillow-SIMD = "^9.0"
torch = [
{markers = "sys_platform == 'linux'", url="https://download.pytorch.org/whl/cu118/torch-2.0.1%2Bcu118-cp310-cp310-linux_x86_64.whl"},
{markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", url="https://download.pytorch.org/whl/cpu/torch-2.0.1-cp310-none-macosx_10_9_x86_64.whl"},
{markers = "sys_platform == 'darwin' and platform_machine == 'arm64'", url="https://download.pytorch.org/whl/cpu/torch-2.0.1-cp310-none-macosx_11_0_arm64.whl"},
{markers = "sys_platform == 'win32'", url="https://download.pytorch.org/whl/cu118/torch-2.0.1%2Bcu118-cp310-cp310-win_amd64.whl"},
]
torchvision = [
{markers = "sys_platform == 'linux'", url="https://download.pytorch.org/whl/cu118/torchvision-0.15.2%2Bcu118-cp310-cp310-linux_x86_64.whl"},
{markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", url="https://download.pytorch.org/whl/cpu/torchvision-0.15.2-cp310-cp310-macosx_10_9_x86_64.whl"},
{markers = "sys_platform == 'darwin' and platform_machine == 'arm64'", url="https://download.pytorch.org/whl/cpu/torchvision-0.15.2-cp310-cp310-macosx_10_9_x86_64.whl"},
{markers = "sys_platform == 'win32'", url="https://download.pytorch.org/whl/cu118/torchvision-0.15.2%2Bcu118-cp310-cp310-win_amd64.whl"},
]
lightning = "^2.0.4"
pydantic = "<2.0.0" # https://github.com/Lightning-AI/lightning/issues/18027
torchmetrics = "^0.11.1"
torchsummary = "^1.5.1"
datasets = "^2.13"
transformers = "^4.30"
accelerate = "^0.20.1"
matplotlib = "^3.6"
numpy = "^1.24"
scikit-image = "^0.19"
scikit-learn = ">=1.0"
tensorboard = "^2.3"
minio = "^7.1.15"
h5py = "^3.9.0"
kornia = "^0.7.0"
gin-config = "^0.5.0"
[tool.poetry.group.test.dependencies]
pytest = "^7.4.0"
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"
# optional scripts
[tool.poetry.scripts]
docmae-train = "docmae.train:main"
docmae-pretrain = "docmae.pretrain_pl:main"
docmae-finetune = "docmae.fine_tune:main"
[tool.black]
line-length = 128
include = '\.pyi?$'
exclude = '''
(
/(
\.eggs # exclude a few common directories in the
| \.git # root of the project
| \.hg
| \.mypy_cache
| \.pytest_cache
| \.tox
| \.venv
| venv
| _build
| buck-out
| build
| dist
)/
)
'''