diff --git a/roop/core.py b/roop/core.py index b667a97..22113c2 100755 --- a/roop/core.py +++ b/roop/core.py @@ -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...")