改用修饰器

GPT_SoVITS/TTS_infer_pack/TTS.py

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