blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-09 00:10:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

改用修饰器

Browse Source
This commit is contained in:
XTer 2024-04-06 23:51:47 +08:00
parent 6591e86df3
commit 7f8892d004
1 changed files with 223 additions and 223 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
GPT_SoVITS/TTS_infer_pack/TTS.py
View File

@ -569,7 +569,8 @@ class TTS:
''' '''
self.stop_flag = True self.stop_flag = True
# 使用装饰器
@torch.no_grad()
def run(self, inputs:dict): def run(self, inputs:dict):
""" """
Text to speech inference. Text to speech inference.
@ -600,8 +601,7 @@ class TTS:
# 直接给全体套一个torch.no_grad()
with torch.no_grad():
########## variables initialization ########### ########## variables initialization ###########
self.stop_flag:bool = False self.stop_flag:bool = False
text:str = inputs.get("text", "") text:str = inputs.get("text", "")