FaceFusion / app.py
Nick088's picture
Fix a typo “cuda” when it should be “conda”
0410452 verified
raw
history blame
619 Bytes
import subprocess
import os
import torch
if torch.cuda.is_available():
device="cuda"
print("Using GPU")
else:
device="cpu"
print("Using CPU")
# Clone the repository
subprocess.run(["git", "clone", "https://github.com/facefusion/facefusion", "--branch", "2.5.0", "--single-branch"], check=True)
# chande directory to face fusion to run ui
os.chdir("facefusion")
# installation
subprocess.run(["python", "install.py", "--onnxruntime", "cuda-11.8", "--skip-conda"], check=True)
# Run the ui
subprocess.run(["python", "run.py", "--execution-providers", "cpu", {"cuda" if device=="cuda" else None}], check=True)