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

check torch gpu before clearing cache

Browse Source
This commit is contained in:
Phan Tuấn Anh 2023-06-05 12:21:37 +02:00 committed by GitHub
parent c59400f6a6
commit 3668360520
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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,7 +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)
torch.cuda.empty_cache()
if torch.cuda.is_available() and args.gpu_vendor in ['amd', 'nvidia']:
torch.cuda.empty_cache()
status("creating video...")
create_video(video_name, exact_fps, output_dir)
status("adding audio...")