mirror of
https://github.com/storytold/FineTrainers-Conditioning.git
synced 2026-10-28 08:57:10 +00:00
Fixing weights copy.
This commit is contained in:
@@ -5,8 +5,8 @@ import torch
|
|||||||
import torch.nn as nn
|
import torch.nn as nn
|
||||||
import torch.nn.functional as F
|
import torch.nn.functional as F
|
||||||
from diffusers import LTXVideoTransformer3DModel
|
from diffusers import LTXVideoTransformer3DModel
|
||||||
from torch.utils import maybe_allow_in_graph
|
from diffusers.utils.torch_utils import maybe_allow_in_graph
|
||||||
from diffusers.models import Transformer2DModelOutput
|
from diffusers.models.modeling_outputs import Transformer2DModelOutput
|
||||||
from diffusers.utils import is_torch_version
|
from diffusers.utils import is_torch_version
|
||||||
from conditioned_residual_adapter_bottleneck import ConditionedResidualAdapterBottleneck
|
from conditioned_residual_adapter_bottleneck import ConditionedResidualAdapterBottleneck
|
||||||
@maybe_allow_in_graph
|
@maybe_allow_in_graph
|
||||||
|
|||||||
Reference in New Issue
Block a user