update code

app.py

@@ -449,7 +449,7 @@ def duplicate_batch(tensor, batch_size=2):
         return None  # 如果是 None，则直接返回
     return tensor.repeat(batch_size, *([1] * (tensor.dim() - 1)))  # 复制 batch 维度
 
-
+@torch.no_grad()
 @spaces.GPU(duration=200)
 def avatar_generation(items, save_path_base, video_path_input, source_type, is_styled, styled_img):
     """
@@ -478,13 +478,11 @@ def avatar_generation(items, save_path_base, video_path_input, source_type, is_s
     exp_img_base_dir = os.path.join(target_path, 'images512x512')
     motion_base_dir = os.path.join(target_path, 'motions')
     label_file_test = os.path.join(target_path, 'images512x512/dataset_realcam.json')
-    render_model =render_model.to(device)
-    image_encoder = image_encoder.to(device)
-    vae_triplane = vae_triplane.to(device)
-    dinov2 = dinov2.to(device)
-    Faceverse = Faceverse.to(device)
-    clip_image_processor = clip_image_processor.to(device)
-    dino_img_processor = dino_img_processor.to(device)
+    render_model.to(device)
+    image_encoder.to(device)
+    vae_triplane.to(device)
+    dinov2.to(device)
+    Faceverse.to(device)
     ws_avg = ws_avg.to(device)
     DiT_model = DiT_model.to(device)
     if source_type == 'example':
@@ -631,6 +629,7 @@ def assert_input_image(input_image):
         raise gr.Error("No image selected or uploaded!")
 
 @spaces.GPU(duration=100)
+@torch.no_grad()
 def process_image(input_image, source_type, is_style, save_dir):
     """ 🎯 处理 input_image，根据是否是示例图片执行不同逻辑 """
     process_img_input_dir = os.path.join(save_dir, 'input_image')
@@ -648,12 +647,13 @@ def process_image(input_image, source_type, is_style, save_dir):
         return imge_dir, source_type  # 这里替换成 处理用户上传图片的逻辑
 
 @spaces.GPU(duration=100)
+@torch.no_grad()
 def style_transfer(processed_image, style_prompt, cfg, strength, save_base):
     """
     🎭 这个函数用于风格转换
     ✅ 你可以在这里填入你的风格化代码
     """
-    pipeline_sd =pipeline_sd.to(device)
+    pipeline_sd.to(device)
     src_img_pil = Image.open(processed_image)
     img_name = os.path.basename(processed_image)
     save_dir = os.path.join(save_base, 'style_img')