diff --git a/finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py b/finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py
index 62618ff..e13f961 100644
--- a/finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py
+++ b/finetrainers/conditioning/LTXVideoConditionedTransformer3DModel.py
@@ -5,8 +5,8 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 from diffusers import LTXVideoTransformer3DModel
-from torch.utils import maybe_allow_in_graph
-from diffusers.models import Transformer2DModelOutput
+from diffusers.utils.torch_utils import maybe_allow_in_graph
+from diffusers.models.modeling_outputs import Transformer2DModelOutput
 from diffusers.utils import is_torch_version
 from conditioned_residual_adapter_bottleneck import ConditionedResidualAdapterBottleneck
 @maybe_allow_in_graph