automatic/modules/omnigen2/models/transformers/__init__.py

4 lines
103 B
Python

from .transformer_omnigen2 import OmniGen2Transformer2DModel
__all__ = ["OmniGen2Transformer2DModel"]