From bca44f6d7f9978225b9f299a2aef58b62e8690c0 Mon Sep 17 00:00:00 2001 From: CrossProduct Date: Wed, 29 Jan 2025 21:11:55 +0000 Subject: [PATCH] Remove some comments. --- finetrainers/conditioning/conditioned_pipeline.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/finetrainers/conditioning/conditioned_pipeline.py b/finetrainers/conditioning/conditioned_pipeline.py index 0b41be7..b9ac1ad 100644 --- a/finetrainers/conditioning/conditioned_pipeline.py +++ b/finetrainers/conditioning/conditioned_pipeline.py @@ -279,9 +279,9 @@ class LTXConditionedPipeline(LTXPipeline): continue # change this # latent_model_input = torch.cat([latents] * 2) if self.do_classifier_free_guidance else latents - #latent_model_input = latent_model_input.to(prompt_embeds.dtype) + # latent_model_input = latent_model_input.to(prompt_embeds.dtype) - # might be the bug .. + latent_model_input = torch.cat([noisy_latent_tokens] * 2) if self.do_classifier_free_guidance else noisy_latent_tokens latent_model_input = latent_model_input.to(prompt_embeds.dtype)