From ea4f6651d5c6895d426cd05e7a1a2847abcb40c0 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Thu, 30 Nov 2023 23:23:38 +0100
Subject: [PATCH] Fix install script

---
 install.js | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/install.js b/install.js
index 2117123..ded6d1e 100644
--- a/install.js
+++ b/install.js
@@ -14,9 +14,16 @@ function install(kernel)
 	{
 		return 'python install.py --onnxruntime cuda --torch cuda';
 	}
-	if (platform === 'linux' && gpu === 'amd')
+	if (gpu === 'amd')
 	{
-		return 'python install.py --onnxruntime directml --torch rocm';
+		if (platform === 'linux')
+		{
+			return 'python install.py --onnxruntime directml --torch rocm';
+		}
+		if (platform === 'win32')
+		{
+			return 'python install.py --onnxruntime directml --torch cpu';
+		}
 	}
 	return 'python install.py --onnxruntime default --torch cpu';
 }