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

syntax error

Browse Source
This commit is contained in:
Phan Tuấn Anh 2023-06-05 12:29:30 +02:00 committed by GitHub
parent b947603441
commit 71af0fac69
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
roop/core.py
View File

@ -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...")