-
Notifications
You must be signed in to change notification settings - Fork 15
/
pyproject.toml
68 lines (63 loc) · 2.86 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
[project]
name = "inference-core-nodes"
dynamic = ["version"]
description = ""
authors = [
{ name = "Ionite", email = "[email protected]" }
]
readme = "README.md"
requires-python = ">=3.10,<3.12"
dependencies = [
"Pillow",
"opencv-python>=4.7.0.72",
"scipy",
"scikit-image",
"python-dateutil",
"mediapipe",
"svglib",
"fvcore",
"yapf",
"omegaconf",
"ftfy",
"addict",
"yacs",
"trimesh[easy]",
"albumentations",
"scikit-learn",
"diffusers>=0.25.0"
]
[project.optional-dependencies]
cpu = []
cuda = [
"onnxruntime-gpu"
]
cuda-12 = [
# CUDA 12, Python 3.10, Windows
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/2692857e-05ef-43b4-ba9c-ccf1c22c437c/_packaging/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/download/onnxruntime-gpu/1.17.1/onnxruntime_gpu-1.17.1-cp310-cp310-win_amd64.whl ; platform_system == 'Windows' and python_version == '3.10'",
# CUDA 12, Python 3.10, Linux
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/2692857e-05ef-43b4-ba9c-ccf1c22c437c/_packaging/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/download/onnxruntime-gpu/1.17.1/onnxruntime_gpu-1.17.1-cp310-cp310-manylinux_2_28_x86_64.whl ; platform_system == 'Linux' and python_version == '3.10'",
# CUDA 12, Python 3.11, Windows
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/2692857e-05ef-43b4-ba9c-ccf1c22c437c/_packaging/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/download/onnxruntime-gpu/1.17.1/onnxruntime_gpu-1.17.1-cp311-cp311-win_amd64.whl ; platform_system == 'Windows' and python_version == '3.11'",
# CUDA 12, Python 3.11, Linux
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/2692857e-05ef-43b4-ba9c-ccf1c22c437c/_packaging/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/download/onnxruntime-gpu/1.17.1/onnxruntime_gpu-1.17.1-cp311-cp311-manylinux_2_28_x86_64.whl ; platform_system == 'Linux' and python_version == '3.11'",
# CUDA 12, Python 3.12, Windows
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/2692857e-05ef-43b4-ba9c-ccf1c22c437c/_packaging/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/download/onnxruntime-gpu/1.17.1/onnxruntime_gpu-1.17.1-cp312-cp312-win_amd64.whl ; platform_system == 'Windows' and python_version == '3.12'",
# CUDA 12, Python 3.12, Linux
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/2692857e-05ef-43b4-ba9c-ccf1c22c437c/_packaging/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/download/onnxruntime-gpu/1.17.1/onnxruntime_gpu-1.17.1-cp312-cp312-manylinux_2_28_x86_64.whl ; platform_system == 'Linux' and python_version == '3.12'"
]
directml = [
"onnxruntime-directml"
]
rocm = [
"onnxruntime-gpu"
]
[tool.setuptools.dynamic]
version = {attr = "inference.__version__"}
[tool.setuptools.packages.find]
where = ["src/"]
include = [
"inference_core_nodes.*"
]
[build-system]
requires = ["flit_core >=3.5,<4"]
build-backend = "flit_core.buildapi"