From c938679de7b87b4f14894d9f57fe0f40dd6e3c06 Mon Sep 17 00:00:00 2001
From: Jairo Correa <jn.j41r0@gmail.com>
Date: Wed, 28 Sep 2022 22:14:13 -0300
Subject: Fix memory leak and reduce memory usage

---
 modules/devices.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'modules/devices.py')

diff --git a/modules/devices.py b/modules/devices.py
index 07bb2339..df63dd88 100644
--- a/modules/devices.py
+++ b/modules/devices.py
@@ -1,4 +1,5 @@
 import torch
+import gc
 
 # has_mps is only available in nightly pytorch (for now), `getattr` for compatibility
 from modules import errors
@@ -17,8 +18,8 @@ def get_optimal_device():
 
     return cpu
 
-
 def torch_gc():
+    gc.collect()
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
         torch.cuda.ipc_collect()
-- 
cgit v1.2.1