From 3c605a124e141143427dc7b34e7e017c25906794 Mon Sep 17 00:00:00 2001 From: CrossProduct Date: Sat, 18 Jan 2025 06:47:39 +0000 Subject: [PATCH] Adding in the conditioned transformer 3d model --- .../conditioning/LTXVideoConditionedTransformer3DModel.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py b/finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py index 704aabe..2fcce45 100644 --- a/finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py +++ b/finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py @@ -80,9 +80,10 @@ def forward( encoder_attention_mask = encoder_attention_mask.unsqueeze(1) batch_size = hidden_states.size(0) - + # inject the condition and the residual then project it into the pretrained proj_in - hidden_states = self.adapter(residual_x=residual_x,conditioned_x=hidden_states) + hidden_states = self.adapter(residual_x=residual_x, + conditioned_x=hidden_states) hidden_states = self.proj_in(hidden_states)