fix pausing

TNTwise · Dec 7, 2024 · 78407dc · 78407dc
1 parent 13f7b5c
commit 78407dc
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/backend/src/pytorch/InterpolateTorch.py b/backend/src/pytorch/InterpolateTorch.py
@@ -50,11 +50,11 @@ def copyTensor(self, tensorToCopy: torch.Tensor, tensorCopiedTo: torch.Tensor):
         self.stream.synchronize()
 
     def hotUnload(self):
-        del self.flownet
-        del self.encode
-        del self.tenFlow_div
-        del self.backwarp_tenGrid
-        del self.f0encode
+        self.flownet = None
+        self.encode = None
+        self.tenFlow_div = None
+        self.backwarp_tenGrid = None
+        self.f0encode = None
         gc.collect()
         torch.cuda.empty_cache()
         torch.cuda.reset_max_memory_allocated()