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)