CalvinBackup/facefusion-pinokio
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-pinokio.git synced 2026-07-23 06:20:48 +02:00
Code Issues Packages Projects Releases Wiki Activity

Remove torch from install.js

Browse Source
This commit is contained in:
henryruhs
2024-03-14 20:44:26 +01:00
parent cc9d7dd821
commit 9eb64f2a5b
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
install.js
+5 -5
View File
@@ -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 =>