Update pipeline_hunyuan_video.py

2025-01-31 23:50:25 +02:00
parent 0b9beb1884
commit 43cb25a038
1 changed files with 5 additions and 1 deletions
@@ -808,9 +808,13 @@ class HunyuanVideoPipeline(DiffusionPipeline):
                    if progress_bar is not None:
                        progress_bar.update()
                    if callback is not None:
+                        if leapfusion_img2vid:
+                            callback_latent = (latent_model_input[:, :, 1:, :, :] - noise_pred[:, :, 1:, :, :] * t / 1000).detach()[0].permute(1,0,2,3)
+                        else:
+                            callback_latent = (latent_model_input - noise_pred * t / 1000).detach()[0].permute(1,0,2,3)
                        callback(
                            i, 
-                            (latent_model_input - noise_pred * t / 1000).detach()[0].permute(1,0,2,3),
+                            callback_latent,
                            None, 
                            num_inference_steps
                        )