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

Fix install script

Browse Source
This commit is contained in:
henryruhs
2023-11-30 23:15:08 +01:00
parent b4540a751f
commit 91040ec2f2
1 changed files with 3 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
install.js
+3 -8
View File
@@ -10,13 +10,13 @@ function install(kernel)
}
return 'python install.py --onnxruntime default --torch default';
}
if (gpu === 'nvidia')
if ([ 'linux', 'win32' ].includes(platform) && gpu === 'nvidia')
{
return 'python install.py --onnxruntime cuda --torch cuda';
}
if (gpu === 'amd')
if (platform === 'linux' && gpu === 'amd')
{
return 'python install.py --onnxruntime default --torch cpu';
return 'python install.py --onnxruntime directml --torch rocm';
}
return 'python install.py --onnxruntime default --torch cpu';
}
@@ -31,14 +31,9 @@ module.exports = async kernel =>
type: 'conda',
name:
[
'cmake',
'ffmpeg'
],
args: '-c conda-forge'
},
{
gpu: 'nvidia',
name: 'cuda'
}
],
run: