From 93f3ed98955dff24b0ba8a582d7ba22dc7ef7aa5 Mon Sep 17 00:00:00 2001
From: anxu <907333918@qq.com>
Date: Wed, 14 Aug 2024 18:09:52 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8E=A8=E7=90=86=E4=B8=8D=E9=9C=80=E8=A6=81?=
 =?UTF-8?q?=E8=AE=A1=E7=AE=97=E6=A2=AF=E5=BA=A6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 musereal.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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
+