Spaces:
Paused
Paused
Update facefusion/installer.py
Browse files- facefusion/installer.py +4 -0
facefusion/installer.py
CHANGED
|
@@ -21,6 +21,10 @@ ONNXRUNTIMES : Dict[str, Tuple[str, str]] =\
|
|
| 21 |
{
|
| 22 |
'default': ('onnxruntime', '1.16.3')
|
| 23 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
if platform.system().lower() == 'linux' or platform.system().lower() == 'windows':
|
| 25 |
TORCH['cuda'] = 'cu118'
|
| 26 |
TORCH['cuda-nightly'] = 'cu121'
|
|
|
|
| 21 |
{
|
| 22 |
'default': ('onnxruntime', '1.16.3')
|
| 23 |
}
|
| 24 |
+
# Hardcoding GPU configurations
|
| 25 |
+
torch = 'cuda'
|
| 26 |
+
onnxruntime = 'cuda'
|
| 27 |
+
|
| 28 |
if platform.system().lower() == 'linux' or platform.system().lower() == 'windows':
|
| 29 |
TORCH['cuda'] = 'cu118'
|
| 30 |
TORCH['cuda-nightly'] = 'cu121'
|