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

clear CUDA cache after swapping

Browse Source 
on low VRAM + ffmpeg cuda acceleration, clearing cache prevent cuda out-of-memory error
This commit is contained in:
Phan Tuấn Anh 2023-06-05 11:58:24 +02:00 committed by GitHub
parent 6daee96029
commit c59400f6a6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
roop/core.py
View File

@ -209,6 +209,7 @@ def start(preview_callback = None):
process_video_multi_cores(args.source_img, args.frame_paths) process_video_multi_cores(args.source_img, args.frame_paths)
else: else:
process_video(args.source_img, args.frame_paths) process_video(args.source_img, args.frame_paths)
torch.cuda.empty_cache()
status("creating video...") status("creating video...")
create_video(video_name, exact_fps, output_dir) create_video(video_name, exact_fps, output_dir)
status("adding audio...") status("adding audio...")