From 9583a6b58a00b5b453790e2971620700810ff8c9 Mon Sep 17 00:00:00 2001
From: Elijah Harmon <elijahharmon@gmail.com>
Date: Wed, 15 Nov 2023 08:18:34 -0800
Subject: [PATCH] won't use cp unless using nvidia

---
 main_onnx.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main_onnx.py b/main_onnx.py
index 6d3dee3..61106bb 100644
--- a/main_onnx.py
+++ b/main_onnx.py
@@ -1,6 +1,5 @@
 import onnxruntime as ort
 import numpy as np
-import cupy as cp
 import gc
 import numpy as np
 import cv2
@@ -32,6 +31,7 @@ def main():
     elif onnxChoice == 2:
         onnxProvider = "DmlExecutionProvider"
     elif onnxChoice == 3:
+        import cupy as cp
         onnxProvider = "CUDAExecutionProvider"
 
     so = ort.SessionOptions()