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

Adding in the conditioned transformer 3d model

Browse Source
This commit is contained in:
CrossProduct
2025-01-18 06:47:39 +00:00
parent fa422c4ca5
commit 3c605a124e
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py
+3 -2
View File
@@ -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)