storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-21 07:55:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Noted place that requires intervention.

Browse Source
This commit is contained in:
CrossProduct
2025-01-29 05:30:17 +00:00
parent 707e001f7b
commit a16285141e
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/conditioning/conditioned_pipeline.py
+3 -3
View File
@@ -279,16 +279,16 @@ 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 = noisy_latent_tokens
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)
# broadcast to batch dimension in a way that's compatible with ONNX/Core ML
timestep = t.expand(latent_model_input.shape[0])
#encoder hidden states are different...
noise_pred = self.transformer(
hidden_states=condition_tokens,
encoder_hidden_states=prompt_embeds,