From 57271ad12546276d6dcbc0bca4fe83d85a1cde43 Mon Sep 17 00:00:00 2001
From: Brandon Rising <brandon@invoke.ai>
Date: Thu, 27 Jul 2023 10:28:26 -0400
Subject: [PATCH] Move onnx to optional dependencies

---
 pyproject.toml | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 362b3b6041..fbe93620a6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -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]