storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-29 09:07:10 +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
+2 -1
View File
@@ -82,7 +82,8 @@ def forward(
batch_size = hidden_states.size(0) batch_size = hidden_states.size(0)
# inject the condition and the residual then project it into the pretrained proj_in # 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) hidden_states = self.proj_in(hidden_states)