1

app.py

@@ -280,6 +280,14 @@ def gradio_infer(
         log = f"Args: {' '.join(args_list)}\n\n{buf.getvalue()}\n\nERROR: {e}"
         return None, log
 
+    # 在合成 mp4 之前加：
+    torch.cuda.synchronize()
+    try:
+        del network, processor, loader, vid_reader, data, rgb, msk, prob
+    except Exception:
+        pass
+    torch.cuda.empty_cache()
+
     # 9) 合成 mp4：从 output/<stem>/ 帧合成 -> TEMP_ROOT/<stem>.mp4
     out_frames = path.join(output_root, video_stem)
     if not path.isdir(out_frames):