From f4e818305bdedd0346df291a36dc065be29303a3 Mon Sep 17 00:00:00 2001
From: SolitaryThinker <wlsaidhi@gmail.com>
Date: Wed, 13 Aug 2025 03:11:22 +0000
Subject: [PATCH] fix

---
 fastvideo/training/distillation_pipeline.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fastvideo/training/distillation_pipeline.py b/fastvideo/training/distillation_pipeline.py
index e7314502c..b97c4223b 100644
--- a/fastvideo/training/distillation_pipeline.py
+++ b/fastvideo/training/distillation_pipeline.py
@@ -412,7 +412,7 @@ class DistillationPipeline(TrainingPipeline):
         training_batch.dmd_latent_vis_dict[
             "generator_timestep"] = target_timestep.float().detach()
 
-        return latents
+        return pred_video
 
     def _dmd_forward(self, generator_pred_video: torch.Tensor,
                      training_batch: TrainingBatch) -> torch.Tensor: