Files
mlx-video/mlx_video/__init__.py

15 lines
400 B
Python

from mlx_video.models.ltx import LTXModel, LTXModelConfig
from mlx_video.models.wan import WanModel, WanModelConfig
from mlx_video.convert import load_transformer_weights, load_vae_weights
import os
__all__ = [
"LTXModel",
"LTXModelConfig",
"WanModel",
"WanModelConfig",
"load_transformer_weights",
"load_vae_weights",
]
os.environ["TRANSFORMERS_NO_ADVISORY_WARNINGS"] = "1"