推理不需要计算梯度

musereal.py

@@ -46,7 +46,7 @@ def __mirror_index(size, index):
         return res
     else:
         return size - res - 1 
-
+@torch.no_grad()
 def inference(render_event,batch_size,latents_out_path,audio_feat_queue,audio_out_queue,res_frame_queue,
               ): #vae, unet, pe,timesteps
     
@@ -320,4 +320,4 @@ class MuseReal(BaseReal):
             #     time.sleep(delay)
         self.render_event.clear() #end infer process render
         print('musereal thread stop')
-            
+