Move onnx to optional dependencies

This commit is contained in:
Brandon Rising 2023-07-27 10:28:26 -04:00
parent 33245b37ad
commit 57271ad125

View File

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