From 09e3f3dd5feee55037eb76160c9178f60115e73e Mon Sep 17 00:00:00 2001 From: CrossProduct Date: Thu, 30 Jan 2025 00:03:12 +0000 Subject: [PATCH] move residual arg. --- .../conditioning/LTXVideoConditionedTransformer3DModel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py b/finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py index d4e5f97..80334dc 100644 --- a/finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py +++ b/finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py @@ -85,13 +85,13 @@ class LTXVideoConditionedTransformer3DModel(LTXVideoTransformer3DModel): encoder_hidden_states: torch.Tensor, timestep: torch.LongTensor, encoder_attention_mask: torch.Tensor, - residual_x: torch.Tensor, num_frames: int, height: int, width: int, rope_interpolation_scale: Optional[Tuple[float, float, float]] = None, attention_kwargs: Optional[Dict[str, Any]] = None, - return_dict: bool = True + return_dict: bool = True, + residual_x: torch.Tensor = None ) -> torch.Tensor: if residual_x == None: print("Residual Not included in the calculation.")