diff --git a/roop/core.py b/roop/core.py
index 22113c2..f392a2a 100755
--- a/roop/core.py
+++ b/roop/core.py
@@ -209,7 +209,7 @@ 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 torch.cuda.is_available() and args.gpu_vendor in ['amd', 'nvidia']:
+    if args.gpu_vendor = 'nvidia':
         torch.cuda.empty_cache() 
     status("creating video...")
     create_video(video_name, exact_fps, output_dir)