ckpt

2025-06-28 14:56:21 -07:00
1 changed files with 1 additions and 0 deletions
@@ -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