diff --git a/musereal.py b/musereal.py
index 9e3d3b9..89ded3c 100644
--- a/musereal.py
+++ b/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')
-            
\ No newline at end of file
+