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

Adjust comment

Browse Source
This commit is contained in:
henryruhs 2023-06-05 12:42:24 +02:00
parent 3747fc2301
commit b116f2001a
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
roop/core.py
View File

@ -209,8 +209,9 @@ def start(preview_callback = None):
process_video_multi_cores(args.source_img, args.frame_paths)
else:
process_video(args.source_img, args.frame_paths)
# prevent out of memory while using ffmpeg with cuda
if args.gpu_vendor == 'nvidia':
torch.cuda.empty_cache() # prevent CUDA OOM when using ffmpeg cuda accel
torch.cuda.empty_cache()
status("creating video...")
create_video(video_name, exact_fps, output_dir)
status("adding audio...")