Call garbage collector after unload model

2022-12-16 16:27:18 +01:00 · 2022-12-16 16:27:18 +01:00 · e5acbc6be9
commit e5acbc6be9
parent 0c4de4776e
1 changed files with 1 additions and 1 deletions
--- a/lama_cleaner/model_manager.py
+++ b/lama_cleaner/model_manager.py
@ -46,8 +46,8 @@ class ModelManager:
            if (torch.cuda.memory_allocated() > 0):
                # Clear current loaded model from memory
                torch.cuda.empty_cache()
-                gc.collect()
                del self.model
+                gc.collect()

            self.model = self.init_model(new_name, self.device, **self.kwargs)
            self.name = new_name