From 71af0fac69deeec3612ceb9cabaa04cbd1e381d9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Phan=20Tu=E1=BA=A5n=20Anh?= Date: Mon, 5 Jun 2023 12:29:30 +0200 Subject: [PATCH] syntax error --- roop/core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/roop/core.py b/roop/core.py index f392a2a..7a48f96 100755 --- a/roop/core.py +++ b/roop/core.py @@ -209,8 +209,8 @@ def start(preview_callback = None): process_video_multi_cores(args.source_img, args.frame_paths) else: process_video(args.source_img, args.frame_paths) - if args.gpu_vendor = 'nvidia': - torch.cuda.empty_cache() + if args.gpu_vendor == 'nvidia': + torch.cuda.empty_cache() # prevent CUDA OOM when using ffmpeg cuda accel status("creating video...") create_video(video_name, exact_fps, output_dir) status("adding audio...")