SadTalker

Runtime error

vinthony commited on Mar 29, 2023

Commit

c17e787

1 Parent(s): f12864f

Update modules/sadtalker_test.py

Files changed (1) hide show

modules/sadtalker_test.py CHANGED Viewed

@@ -106,6 +106,7 @@ class SadTalker():
         torch.cuda.empty_cache()
         torch.cuda.synchronize()
         if use_enhancer:
             return os.path.join(save_dir, video_name+'_enhanced.mp4'), os.path.join(save_dir, video_name+'_enhanced.mp4')

         torch.cuda.empty_cache()
         torch.cuda.synchronize()
+        import gc; gc.collect()
         if use_enhancer:
             return os.path.join(save_dir, video_name+'_enhanced.mp4'), os.path.join(save_dir, video_name+'_enhanced.mp4')