From 8aa849e270f6ccfb16a13894ecbf9ae09a7bf8af Mon Sep 17 00:00:00 2001
From: ylzz1997 <ylzz1997@outlook.com>
Date: Wed, 12 Apr 2023 00:44:16 +0800
Subject: [PATCH] Debug

---
 inference/infer_tool.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/inference/infer_tool.py b/inference/infer_tool.py
index 6a2f4df..94b3ca6 100644
--- a/inference/infer_tool.py
+++ b/inference/infer_tool.py
@@ -6,6 +6,7 @@ import os
 import time
 from pathlib import Path
 from inference import slicer
+import gc
 
 import librosa
 import numpy as np
@@ -223,8 +224,13 @@ class Svc(object):
 
     def unload_model(self):
         # 卸载模型
+        self.net_g_ms = self.net_g_ms.to("cpu")
         del self.net_g_ms
-        if hasattr(self,"enhancer"): del self.enhancer
+        if hasattr(self,"enhancer"): 
+            self.enhancer.enhancer = self.enhancer.enhancer.to("cpu")
+            del self.enhancer.enhancer
+            del self.enhancer
+        gc.collect()
 
     def slice_inference(self,
                         raw_audio_path,