ProPainter12

Paused

goryhon commited on Apr 11

Commit

f3aed33

verified ·

1 Parent(s): 444f7ef

Update web-demos/hugging_face/tracker/base_tracker.py

Files changed (1) hide show

web-demos/hugging_face/tracker/base_tracker.py CHANGED Viewed

@@ -100,4 +100,5 @@ class BaseTracker:
     def clear_memory(self):
         self.tracker.clear_memory()
         self.mapper.clear_labels()
-        torch.cuda.empty_cache()

     def clear_memory(self):
         self.tracker.clear_memory()
         self.mapper.clear_labels()
+        torch.cuda.empty_cache()
+        torch.cuda.ipc_collect()