from .DiT import DiT_models, DiT
from .Transformer import Transformer_models, Transformer
from .EMA import EMAModel

All_models = {**DiT_models, **Transformer_models}