From c3da64b0ffbe4adf2c461c0ceb1cc6957554dc1c Mon Sep 17 00:00:00 2001 From: CyberWon Date: Sun, 23 Jun 2024 12:43:10 +0800 Subject: [PATCH] =?UTF-8?q?Optimize=EF=BC=9A=20=E6=8E=A8=E7=90=86=E5=AE=8C?= =?UTF-8?q?=E6=88=90=E5=90=8E=E8=87=AA=E5=8A=A8=E8=A7=A6=E5=8F=91gc?= =?UTF-8?q?=E5=9B=9E=E6=94=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 在推理完成之后，使用gc控制器对内存进行回收。 #1218 --- GPT_SoVITS/TTS_infer_pack/TTS.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/GPT_SoVITS/TTS_infer_pack/TTS.py b/GPT_SoVITS/TTS_infer_pack/TTS.py index eaacb529..78bd9a82 100644 --- a/GPT_SoVITS/TTS_infer_pack/TTS.py +++ b/GPT_SoVITS/TTS_infer_pack/TTS.py @@ -1,6 +1,6 @@ from copy import deepcopy import math -import os, sys +import os, sys, gc import random import traceback @@ -865,7 +865,8 @@ class TTS: self.empty_cache() def empty_cache(self): - try: + try: + gc.collect() # 触发gc的垃圾回收。避免内存一直增长。 if "cuda" in str(self.configs.device): torch.cuda.empty_cache() elif str(self.configs.device) == "mps": @@ -935,4 +936,4 @@ def speed_change(input_audio:np.ndarray, speed:float, sr:int): # 将管道输出解码为 NumPy 数组 processed_audio = np.frombuffer(out, np.int16) - return processed_audio \ No newline at end of file + return processed_audio