From f4ffd1c64bbea55b875b3a2550d015f5c1acb1c3 Mon Sep 17 00:00:00 2001
From: bubbliiiing <3323290568@qq.com>
Date: Thu, 31 Jul 2025 15:55:06 +0800
Subject: [PATCH] Update Wan2.2 Speed

---
 videox_fun/pipeline/pipeline_wan2_2.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/videox_fun/pipeline/pipeline_wan2_2.py b/videox_fun/pipeline/pipeline_wan2_2.py
index ad0f0b2..58e12f2 100755
--- a/videox_fun/pipeline/pipeline_wan2_2.py
+++ b/videox_fun/pipeline/pipeline_wan2_2.py
@@ -578,6 +578,7 @@ class Wan2_2Pipeline(DiffusionPipeline):
                         progress_bar.update()
                     if comfyui_progressbar:
                         pbar.update(1)
+            return latents
 
         latents = dit_forward(latents)