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)