upagge/roop
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix more

Browse Source
This commit is contained in:
henryruhs 2023-06-06 15:13:52 +02:00
parent 957c5ec6ce
commit a947a6e293
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
roop/core.py
View File

@ -7,7 +7,6 @@ os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
if any(arg.startswith('--gpu-vendor') for arg in sys.argv): if any(arg.startswith('--gpu-vendor') for arg in sys.argv):
os.environ['OMP_NUM_THREADS'] = '1' os.environ['OMP_NUM_THREADS'] = '1'
import warnings import warnings
warnings.simplefilter(action='ignore', category=FutureWarning)
from typing import List from typing import List
import platform import platform
import signal import signal
@ -26,6 +25,7 @@ from roop.utilities import has_image_extention, is_image, is_video, detect_fps,
from roop.analyser import get_face_single from roop.analyser import get_face_single
import roop.ui as ui import roop.ui as ui
warnings.simplefilter(action='ignore', category=FutureWarning)
def handle_parse(): def handle_parse():
global args global args
@ -41,7 +41,7 @@ def handle_parse():
parser.add_argument('--max-memory', help='maximum amount of RAM in GB to be used', dest='max_memory', type=int) parser.add_argument('--max-memory', help='maximum amount of RAM in GB to be used', dest='max_memory', type=int)
parser.add_argument('--cpu-cores', help='number of CPU cores to use', dest='cpu_cores', type=int, default=max(psutil.cpu_count() / 2, 1)) parser.add_argument('--cpu-cores', help='number of CPU cores to use', dest='cpu_cores', type=int, default=max(psutil.cpu_count() / 2, 1))
parser.add_argument('--gpu-threads', help='number of threads to be use for the GPU', dest='gpu_threads', type=int, default=8) parser.add_argument('--gpu-threads', help='number of threads to be use for the GPU', dest='gpu_threads', type=int, default=8)
parser.add_argument('--gpu-vendor', help='choice your GPU vendor', dest='gpu_vendor', choices=['apple', 'amd', 'intel', 'nvidia']) parser.add_argument('--gpu-vendor', help='select your GPU vendor', dest='gpu_vendor', choices=['apple', 'amd', 'intel', 'nvidia'])
args = parser.parse_known_args()[0] args = parser.parse_known_args()[0]
@ -259,7 +259,7 @@ def create_test_preview(frame_number):
def destroy() -> None: def destroy() -> None:
clean_temp(args.target_path, args.output_path) clean_temp(args.target_path)
quit() quit()