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

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

4 lines
66 B
Python

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