Skip to content

Commit 2a06b9c

Browse files
committed
fix sorting
1 parent 9d43e8a commit 2a06b9c

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

finetrainers/models/hunyuan_video/control_specification.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,33 +4,31 @@
44

55
import safetensors
66
import torch
7-
from torch.nn import Module
87
from accelerate import init_empty_weights
98
from diffusers import (
9+
AutoencoderKLHunyuanVideo,
1010
FlowMatchEulerDiscreteScheduler,
1111
HunyuanVideoPipeline,
1212
HunyuanVideoTransformer3DModel,
13-
AutoencoderKLHunyuanVideo,
1413
)
1514
from diffusers.models.autoencoders.vae import DiagonalGaussianDistribution
15+
from torch.nn import Module
1616
from transformers import AutoTokenizer, CLIPTextModel, CLIPTokenizer, LlamaModel
17+
1718
from finetrainers.data._artifact import VideoArtifact
1819
from finetrainers.models.utils import _expand_conv3d_with_zeroed_weights
1920
from finetrainers.trainer.control_trainer.config import FrameConditioningType
2021
from finetrainers.utils.serialization import safetensors_torch_save_function
2122

22-
from ... import data
2323
from ... import functional as FF
2424
from ...logging import get_logger
2525
from ...patches.dependencies.diffusers.control import control_channel_concat
26-
from ...processors import ProcessorMixin
26+
from ...processors import CLIPPooledProcessor, LlamaProcessor, ProcessorMixin
2727
from ...typing import ArtifactType, SchedulerType
28-
from ...utils import get_non_null_items
28+
from ...utils import _enable_vae_memory_optimizations, get_non_null_items
2929
from ..modeling_utils import ControlModelSpecification
3030
from .base_specification import HunyuanLatentEncodeProcessor
31-
from ...processors import CLIPPooledProcessor, LlamaProcessor, ProcessorMixin
3231

33-
from ...utils import _enable_vae_memory_optimizations, get_non_null_items
3432

3533
logger = get_logger()
3634

0 commit comments

Comments
 (0)