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:23:38 +01:00
parent 91040ec2f2
commit ea4f6651d5
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
install.js
+9 -2
View File
@@ -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';
}