|
4 | 4 |
|
5 | 5 | import safetensors |
6 | 6 | import torch |
7 | | -from torch.nn import Module |
8 | 7 | from accelerate import init_empty_weights |
9 | 8 | from diffusers import ( |
| 9 | + AutoencoderKLHunyuanVideo, |
10 | 10 | FlowMatchEulerDiscreteScheduler, |
11 | 11 | HunyuanVideoPipeline, |
12 | 12 | HunyuanVideoTransformer3DModel, |
13 | | - AutoencoderKLHunyuanVideo, |
14 | 13 | ) |
15 | 14 | from diffusers.models.autoencoders.vae import DiagonalGaussianDistribution |
| 15 | +from torch.nn import Module |
16 | 16 | from transformers import AutoTokenizer, CLIPTextModel, CLIPTokenizer, LlamaModel |
| 17 | + |
17 | 18 | from finetrainers.data._artifact import VideoArtifact |
18 | 19 | from finetrainers.models.utils import _expand_conv3d_with_zeroed_weights |
19 | 20 | from finetrainers.trainer.control_trainer.config import FrameConditioningType |
20 | 21 | from finetrainers.utils.serialization import safetensors_torch_save_function |
21 | 22 |
|
22 | | -from ... import data |
23 | 23 | from ... import functional as FF |
24 | 24 | from ...logging import get_logger |
25 | 25 | from ...patches.dependencies.diffusers.control import control_channel_concat |
26 | | -from ...processors import ProcessorMixin |
| 26 | +from ...processors import CLIPPooledProcessor, LlamaProcessor, ProcessorMixin |
27 | 27 | 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 |
29 | 29 | from ..modeling_utils import ControlModelSpecification |
30 | 30 | from .base_specification import HunyuanLatentEncodeProcessor |
31 | | -from ...processors import CLIPPooledProcessor, LlamaProcessor, ProcessorMixin |
32 | 31 |
|
33 | | -from ...utils import _enable_vae_memory_optimizations, get_non_null_items |
34 | 32 |
|
35 | 33 | logger = get_logger() |
36 | 34 |
|
|
0 commit comments