omni / src /models /lm /__init__.py
chenbhao's picture
Flatten src layout: drop omni namespace, modules live directly under src/
92c7321
Raw
History Blame Contribute Delete
131 Bytes
from models.lm.config import LMConfig
from models.lm.model import LMForCausalLM, LM
__all__ = ["LMConfig", "LMForCausalLM", "LM"]