From 9fde6600dcf760fd65007093fe6a3658ca2e7081 Mon Sep 17 00:00:00 2001 From: CrossProduct Date: Wed, 22 Jan 2025 01:26:17 +0000 Subject: [PATCH] Fixing weights copy. --- .../conditioning/LTXVideoConditionedTransformer3DModel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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