Revert "add cuda empty cache to properly release cuda memory"

This reverts commit c78e8f1.
nadermx · Apr 19, 2024 · 4ba6485 · 4ba6485
1 parent c78e8f1
commit 4ba6485
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/backgroundremover/u2net/detect.py b/backgroundremover/u2net/detect.py
@@ -151,6 +151,5 @@ def predict(net, item):
         img = Image.fromarray(predict_np * 255).convert("RGB")
 
         del d1, d2, d3, d4, d5, d6, d7, pred, predict, predict_np, inputs_test, sample
-        torch.cuda.empty_cache() if torch.cuda.is_available() else None
 
         return img