make residual optional.

This commit is contained in:
CrossProduct
2025-01-30 00:02:31 +00:00
parent 8fe0ed225a
commit be1cb10894
@@ -93,7 +93,8 @@ class LTXVideoConditionedTransformer3DModel(LTXVideoTransformer3DModel):
attention_kwargs: Optional[Dict[str, Any]] = None,
return_dict: bool = True
) -> torch.Tensor:
if residual_x == None:
print("Residual Not included in the calculation.")
image_rotary_emb = self.rope(hidden_states, num_frames, height, width, rope_interpolation_scale)
# convert encoder_attention_mask to a bias the same way we do for attention_mask