forked from neggles/masquerade
-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.cfg
94 lines (87 loc) · 2.03 KB
/
setup.cfg
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
[metadata]
name = masquerade
author = Andi Powers-Holmes
email = [email protected]
maintainer = Andi Powers-Holmes
maintainer_email = [email protected]
license_files = LICENSE.md
[options]
python_requires = >=3.8
packages = find_namespace:
package_dir =
=src
py_modules =
masquerade
include_package_data = True
install_requires =
accelerate >= 0.20.3
crc32c >= 2.3
colorama >= 0.4.3, < 0.5.0
datasets >= 2.13.0
diffusers >= 0.17.1
einops >= 0.6.1
kornia >= 0.6.12
lightning[extra] >= 2.0.0, < 2.1.0
natsort >= 8.4.0
ninja >= 1.11.0
numpy >= 1.24.3
omegaconf >= 2.3.0
open-clip-torch >= 2.20.0
pillow >= 9.5.0, < 10.0.0
pudb >= 2022.1.3
pydantic >= 1.10.8, < 2.0.0
pynvml >= 11.0.0, < 12.0.0
pytorch-fid >= 0.3.0
rich >= 13.0.0, < 14.0.0
safetensors >= 0.3.1
sentencepiece >= 0.1.99
shellingham >= 1.5.0, < 2.0.0
soundfile >= 0.12.1
tensorboardx >= 2.6
tokenizers >= 0.13.3
torch >= 2.0.0, < 2.2.0
torchvision
tqdm >= 4.65.0
transformers >= 4.30.2
typer >= 0.9.0, < 1.0.0
vector-quantize-pytorch >= 1.6.24
wandb >= 0.15.4, < 0.16.0
webdataset >= 0.2.33
[options.packages.find]
where = src
[options.package_data]
* =
*.css
*.json
*.js
*.md
*.txt
masquerade.data =
*.ckpt
*.pt
*.pth
*.safetensors
[options.extras_require]
dev =
black >= 22.3.0
ruff >= 0.0.234
setuptools-scm >= 7.0.0
mypy >= 0.981
[options.entry_points]
console_scripts =
masquerade = masquerade.cli:app
[flake8]
max-line-length = 110
ignore =
# these are annoying during development but should be enabled later
F401 # module imported but unused
F841 # local variable is assigned to but never used
# black automatically fixes this
E501 # line too long
# black breaks these two rules:
E203 # whitespace before :
W503 # line break before binary operator
# this rule makes flake hate Typer so
B008 # do not perform function calls in argument defaults
extend-exclude =
.venv