forked from nikopueringer/CorridorKey
-
Notifications
You must be signed in to change notification settings - Fork 131
Expand file tree
/
Copy pathpyproject.toml
More file actions
69 lines (62 loc) · 1.84 KB
/
Copy pathpyproject.toml
File metadata and controls
69 lines (62 loc) · 1.84 KB
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
[project]
name = "corridorkey"
version = "2.1.5"
description = "AI Chroma Keyer by Corridor Digital"
license = {text = "CC-BY-NC-SA-4.0"}
requires-python = ">=3.10,<3.14"
readme = "README.md"
dependencies = [
"torch==2.9.1",
"torchvision==0.24.1",
"numpy",
"opencv-python-headless>=4.10,<5", # 5.0 wheels dropped OpenEXR (#180)
"OpenEXR>=3.4",
"timm==1.0.24",
"av",
"diffusers",
"easydict",
"einops",
"kornia",
"imageio",
"huggingface_hub",
"socksio", # socks5 proxy support for httpx model downloads
"peft",
"Pillow",
"PIMS",
"setuptools",
"tqdm",
"accelerate",
"transformers",
"matplotlib",
"nvidia-ml-py",
"PySide6>=6.7.0",
"triton-windows>=3.5,<3.6; sys_platform == 'win32'",
"cryptography>=43.0",
]
[project.optional-dependencies]
dev = ["pytest>=7.0", "ruff"]
tracker = ["sam-2 @ git+https://github.com/facebookresearch/sam2.git@2b90b9f5ceec907a1c18123530e92e794ad901a4"]
quality = [
"scikit-image",
"pytorch-msssim",
"lpips",
]
mlx = [
"mlx>=0.30 ; sys_platform == 'darwin' and python_version >= '3.11'",
"safetensors ; sys_platform == 'darwin' and python_version >= '3.11'",
"pyobjc-framework-Vision ; sys_platform == 'darwin' and python_version >= '3.11'",
"pyobjc-framework-Quartz ; sys_platform == 'darwin' and python_version >= '3.11'",
]
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"
[tool.hatch.build.targets.wheel]
packages = ["backend", "ui", "CorridorKeyModule", "gvm_core", "sam2_tracker", "VideoMaMaInferenceModule", "modules/MatAnyone2Module"]
[tool.hatch.metadata]
allow-direct-references = true
[tool.uv]
torch-backend = "auto"
[tool.uv.sources]
corridorkey-mlx = { git = "https://github.com/nikopueringer/corridorkey-mlx.git", tag = "v1.0.0" }
[tool.pytest.ini_options]
testpaths = ["tests"]