Open-Sora/open_sora/modeling/dit/__init__.py

4 lines
66 B
Python
Raw Normal View History

from .dit import DiT, DiT_models
__all__ = ["DiT_models", "DiT"]