syntax error

2023-06-05 12:29:30 +02:00
parent b947603441
commit 71af0fac69
1 changed files with 2 additions and 2 deletions
--- 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...")