MPCompress/pyproject.toml at dev · faymek/MPCompress · GitHub

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
[tool.black]
line-length = 120

[tool.pylint.master]
persistent = false
score = false

[tool.pylint.messages_control]
disable = "all"
enable = [
  "miscellaneous",
  "similarities",
]

[tool.pylint.similarities]
ignore-comments = true
ignore-docstrings = true
ignore-imports = true
min-similarity-lines = 8

[tool.pylint.reports]
reports = false

[tool.pylint.miscellaneous]
notes = [
  "FIXME",
  "XXX",
  "TODO",
]

[tool.poetry]
name = "mpcompress"
version = "0.1.0"
description = ""
authors = ["Faymek Feng <faymek.f@gmail.com>"]
readme = "README.md"

[tool.poetry.dependencies]
python = "^3.10"
torch = {version = "2.4.0", source = "pytorch-cu121"}
torchvision = {version = "0.19.0", source = "pytorch-cu121"}
omegaconf = "^2.3.0"
torchmetrics = "^1.7.0"
iopath = "^0.1.10"
fvcore = "^0.1.5.post20221221"
xformers = "^0.0.27.post2"
# submitit = { git = "https://github.com/facebookincubator/submitit.git" }
tensorboard = "^2.19.0"
timm = "^1.0.15"
pytorch-lightning = "^2.5.1"
pyiqa = "^0.1.13"
transformers = "4.37.2"
diffusers = "^0.33.1"
accelerate = "^1.6.0"
plotly = "^6.0.1"
open-clip-torch = "^2.32.0"
einops = "^0.8.1"
gdown = "^5.2.0"
huggingface-hub = "^0.31.2"
wandb = "^0.19.11"
torch-fidelity = "^0.3.0"
torchinfo = "^1.8.0"
webdataset = "^0.2.111"
notebook = "^7.4.3"
tqdm = "^4.67.1"
pybind11 = "^2.13.6"
compressai = "^1.2.8"
torchac = "^0.9.3"
dotenv = "^0.9.9"

[tool.poetry.group.docs]
optional = true

[tool.poetry.group.docs.dependencies]
zensical = "*"
mkdocstrings-python = "*"

[[tool.poetry.source]]
name = "pytorch-cu121"
url = "https://download.pytorch.org/whl/cu121"
priority = "explicit"

[[tool.poetry.source]]
name = "xformers"
url = "https://pypi.org/simple/"
priority = "explicit"

[[tool.poetry.source]]
name = "nvidia"
url = "https://pypi.nvidia.com"
priority = "explicit"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"