From 9eb64f2a5b63b541dd66c06be32e936d6dc1b0ad Mon Sep 17 00:00:00 2001 From: henryruhs Date: Thu, 14 Mar 2024 20:44:26 +0100 Subject: [PATCH] Remove torch from install.js --- install.js | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/install.js b/install.js index 22ef30e..56b0153 100644 --- a/install.js +++ b/install.js @@ -4,24 +4,24 @@ function install(kernel) if (platform === 'darwin') { - return 'python install.py --onnxruntime default --torch default --skip-venv'; + return 'python install.py --onnxruntime default --skip-venv'; } if ([ 'linux', 'win32' ].includes(platform) && gpu === 'nvidia') { - return 'python install.py --onnxruntime cuda-11.8 --torch cuda-11.8 --skip-venv'; + return 'python install.py --onnxruntime cuda-11.8 --skip-venv'; } if (gpu === 'amd') { if (platform === 'linux') { - return 'python install.py --onnxruntime rocm-5.4.2 --torch rocm-5.4.2 --skip-venv'; + return 'python install.py --onnxruntime rocm-5.4.2 --skip-venv'; } if (platform === 'win32') { - return 'python install.py --onnxruntime directml --torch cpu --skip-venv'; + return 'python install.py --onnxruntime directml --skip-venv'; } } - return 'python install.py --onnxruntime default --torch cpu --skip-venv'; + return 'python install.py --onnxruntime default --skip-venv'; } module.exports = async kernel =>