Refactor Wan model structure by renaming and relocating model imports from model.py to wan2.py, enhancing code organization and clarity across the Wan2 module.
This commit is contained in:
@@ -4,4 +4,4 @@ from mlx_video.models.ltx_2.config import (
|
||||
LTXModelType,
|
||||
TransformerConfig,
|
||||
)
|
||||
from mlx_video.models.ltx_2.ltx import LTXModel, X0Model
|
||||
from mlx_video.models.ltx_2.ltx_2 import LTXModel, X0Model
|
||||
|
||||
@@ -33,7 +33,7 @@ from mlx_video.models.ltx_2.conditioning import (
|
||||
apply_conditioning,
|
||||
)
|
||||
from mlx_video.models.ltx_2.conditioning.latent import LatentState, apply_denoise_mask
|
||||
from mlx_video.models.ltx_2.ltx import LTXModel
|
||||
from mlx_video.models.ltx_2.ltx_2 import LTXModel
|
||||
from mlx_video.models.ltx_2.transformer import Modality
|
||||
from mlx_video.models.ltx_2.upsampler import load_upsampler, upsample_latents
|
||||
from mlx_video.models.ltx_2.video_vae import VideoEncoder
|
||||
|
||||
Reference in New Issue
Block a user