from .dit_3d import DiT_models