From ea8c154176eae549bee71eb772e71fcf07b86bc1 Mon Sep 17 00:00:00 2001 From: SolitaryThinker Date: Sat, 28 Jun 2025 14:56:21 -0700 Subject: [PATCH] ckpt --- fastvideo/v1/pipelines/stages/encoding.py | 1 + 1 file changed, 1 insertion(+) diff --git a/fastvideo/v1/pipelines/stages/encoding.py b/fastvideo/v1/pipelines/stages/encoding.py index 410dc2aee..873523c53 100644 --- a/fastvideo/v1/pipelines/stages/encoding.py +++ b/fastvideo/v1/pipelines/stages/encoding.py @@ -103,6 +103,7 @@ class EncodingStage(PipelineStage): generator = batch.generator if generator is None: raise ValueError("Generator must be provided") + # latent_condition = self.retrieve_latents(encoder_output, generator, sample_mode="argmax") latent_condition = self.retrieve_latents(encoder_output, generator) # Apply shifting if needed