fix

2025-08-13 03:11:22 +00:00
parent 4c4ab65020
commit f4e818305b
1 changed files with 1 additions and 1 deletions
@@ -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: