Update facefusion/installer.py

facefusion/installer.py

@@ -21,9 +21,6 @@ ONNXRUNTIMES : Dict[str, Tuple[str, str]] =\
 {
 	'default': ('onnxruntime', '1.16.3')
 }
-# Hardcoding GPU configurations
-torch = 'cuda'
-onnxruntime = 'cuda'
 
 if platform.system().lower() == 'linux' or platform.system().lower() == 'windows':
 	TORCH['cuda'] = 'cu118'
@@ -56,21 +53,12 @@ def run(program : ArgumentParser) -> None:
 
 	if not args.skip_venv:
 		os.environ['PIP_REQUIRE_VIRTUALENV'] = '1'
-	if args.torch and args.onnxruntime:
-		answers =\
-		{
-			'torch': args.torch,
-			'onnxruntime': args.onnxruntime
-		}
-	else:
-		answers = inquirer.prompt(
-		[
-			inquirer.List('torch', message = wording.get('install_dependency_help').format(dependency = 'torch'), choices = list(TORCH.keys())),
-			inquirer.List('onnxruntime', message = wording.get('install_dependency_help').format(dependency = 'onnxruntime'), choices = list(ONNXRUNTIMES.keys()))
-		])
-	if answers:
-		torch = answers['torch']
-		torch_wheel = TORCH[torch]
+    answers = {
+        'torch': 'gpu',         # Assuming 'yes' for torch GPU configuration
+        'onnxruntime': 'gpu'    # Assuming 'yes' for onnxruntime GPU configuration
+    }
+
+        torch_wheel = TORCH[answers['torch']]
 		onnxruntime = answers['onnxruntime']
 		onnxruntime_name, onnxruntime_version = ONNXRUNTIMES[onnxruntime]