from .latte.modeling_latte import Latte_models

Diffusion_models = {}
Diffusion_models.update(Latte_models)