Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move onnx to optional dependencies

Browse Source
This commit is contained in:
Brandon Rising 2023-07-27 10:28:26 -04:00
parent 33245b37ad
commit 57271ad125
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pyproject.toml
View File

@ -62,8 +62,6 @@ dependencies = [
"npyscreen", "npyscreen",
"numpy<1.24", "numpy<1.24",
"omegaconf", "omegaconf",
"onnx",
"onnxruntime-gpu",
"opencv-python", "opencv-python",
"picklescan", "picklescan",
"pillow", "pillow",
@ -105,6 +103,18 @@ dependencies = [
"xformers~=0.0.19; sys_platform!='darwin'", "xformers~=0.0.19; sys_platform!='darwin'",
"triton; sys_platform=='linux'", "triton; sys_platform=='linux'",
] ]
"onnx" = [
"onnx",
"onnxruntime",
]
"onnx-cuda" = [
"onnx",
"onnxruntime-gpu",
]
"onnx-directml" = [
"onnx",
"onnxruntime-directml",
]
[project.scripts] [project.scripts]