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

move residual arg.

Browse Source
This commit is contained in:
CrossProduct
2025-01-30 00:03:12 +00:00
parent be1cb10894
commit 09e3f3dd5f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py
+2 -2
View File
@@ -85,13 +85,13 @@ class LTXVideoConditionedTransformer3DModel(LTXVideoTransformer3DModel):
encoder_hidden_states: torch.Tensor, encoder_hidden_states: torch.Tensor,
timestep: torch.LongTensor, timestep: torch.LongTensor,
encoder_attention_mask: torch.Tensor, encoder_attention_mask: torch.Tensor,
residual_x: torch.Tensor,
num_frames: int, num_frames: int,
height: int, height: int,
width: int, width: int,
rope_interpolation_scale: Optional[Tuple[float, float, float]] = None, rope_interpolation_scale: Optional[Tuple[float, float, float]] = None,
attention_kwargs: Optional[Dict[str, Any]] = None, attention_kwargs: Optional[Dict[str, Any]] = None,
return_dict: bool = True return_dict: bool = True,
residual_x: torch.Tensor = None
) -> torch.Tensor: ) -> torch.Tensor:
if residual_x == None: if residual_x == None:
print("Residual Not included in the calculation.") print("Residual Not included in the calculation.")