File size: 586 Bytes
962014c
ebc81de
5f091f6
 
 
fb6f00b
5f091f6
 
 
60e78ea
298cb83
 
60e78ea
ebc81de
 
298cb83
5f091f6
 
 
298cb83
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import subprocess
import os
import torch

if torch.cuda.is_available():
  subprocess.run(["apt-get", "install", "nvidia-cuda-toolkit"], check=True)
  print("Using GPU")
else:
  print("Using CPU")

# Clone the repository
subprocess.run(["git", "clone", "https://github.com/facefusion/facefusion"], check=True)

# chande directory to face fusion to run ui
os.chdir("facefusion")

# installation
subprocess.run(["python", "install.py", "--onnxruntime", "cuda-11.8", "--skip-venv"], check=True)

# Run the ui
subprocess.run(["python", "run.py", "--execution-providers", "cpu"], check=True)