Open-Sora/open_sora/modeling/__init__.py
Frank Lee 9648d53d4d
added latte sampling (#22)
* added latte sampling

* polish
2024-03-04 10:43:22 +08:00

5 lines
106 B
Python

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