clear CUDA cache after swapping

on low VRAM + ffmpeg cuda acceleration, clearing cache prevent cuda out-of-memory error
2023-06-05 11:58:24 +02:00 · 2023-06-05 11:58:24 +02:00 · c59400f6a6
commit c59400f6a6
parent 6daee96029
1 changed files with 1 additions and 0 deletions
--- a/roop/core.py
+++ b/roop/core.py
@ -209,6 +209,7 @@ def start(preview_callback = None):
        process_video_multi_cores(args.source_img, args.frame_paths)
    else:
        process_video(args.source_img, args.frame_paths)
+    torch.cuda.empty_cache() 
    status("creating video...")
    create_video(video_name, exact_fps, output_dir)
    status("adding audio...")