-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathpyproject.toml
99 lines (94 loc) · 2.21 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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
[tool.poetry]
name = "wave-image-styler"
version = "0.3.2"
license = "MIT"
description = "App for styling portraits (Portrait Booth) using Machine learning and computer vision"
authors = [
"Pramit Choudhary <[email protected]>",
"Dilith Jayakody <[email protected]>",
"Michal Malohlava <[email protected]>",
"Srinivas Neppalli <[email protected]>"
]
readme = "README.md"
classifiers = [
"Development Status :: 4 - Beta",
"Environment :: Console",
"Intended Audience :: Developers",
"License :: Other/Proprietary License",
"Programming Language :: Python",
"Programming Language :: Python :: 3.8",
"Topic :: Scientific/Engineering",
"Topic :: Scientific/Engineering :: Artificial Intelligence"
]
[tool.poetry.dependencies]
python = ">=3.8.16,<3.10"
h2o-wave = "^0.22.0"
torch = "^2.0.0"
pytorch-lightning ="^1.5.0"
diffusers = "^0.16.0"
torchvision = "^0.15.1"
numpy = "^1.21.6"
Pillow = "^9.0.1"
aiofiles = "^0.8.0"
python-multipart = "^0.0.5"
loguru = "^0.6.0"
opencv-python = "^4.5.5"
pandas = "^1.2.4"
matplotlib = "^3.5.1"
scipy = "^1.4.1"
cmake = "^3.22.2"
dlib = "^19.23.0"
dalle-mini = "^0.1.2"
einops = "^0.6.1"
basicsr = "^1.4.2"
facexlib = "^0.3.0"
requests = "^2.27.1"
deepface = "^0.0.73"
ftfy = "^6.1.1"
transformers = "^4.30.1"
tensorflow = "^2.12.0"
ninja = "^1.10.2"
omegaconf = "^2.3.0"
jax = "0.4.10"
jaxlib = "0.4.10"
realesrgan = "^0.3.0"
toml = "^0.10.2"
open_clip_torch = "^2.0.2"
xformers="^0.0.18"
timm="^0.9.2"
safetensors="^0.3.1"
prettytable="^3.7.0"
accelerate="^0.17.0"
[tool.poetry.dev-dependencies]
pylint = { version = "^2.12.2", allow-prereleases = true }
flake8 = { version = "^4.0.1", allow-prereleases = true }
black = { version = "21.12b0", allow-prereleases = true }
[tool.black]
line-length = 120
skip-string-normalization = true
target-version = ['py38.16']
include = '\.pyi?$'
exclude = '''
(
/(
\.eggs # exclude a few common directories in the
| \.git # root of the project
| \.hg
| \.mypy_cache
| \.tox
| \.venv
| _build
| buck-out
| build
| dist
)/
| notebooks
| local
)
'''
[tool.isort]
line_length = 120
multi_line_output = 3
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"