from .dit.modeling_dit import DiT_models
from .latte.modeling_latte import Latte_models, LatteT2V_models

Diffusion_models = {}
Diffusion_models.update(DiT_models)
Diffusion_models.update(Latte_models)
Diffusion_models.update(LatteT2V_models)