omni / src /trainers /__init__.py
chenbhao's picture
Flatten src layout: drop omni namespace, modules live directly under src/
92c7321
Raw
History Blame Contribute Delete
456 Bytes
from trainers.lm.pretrain import *
from trainers.lm.full_sft import *
from trainers.lm.lora import *
from trainers.lm.dpo import *
from trainers.lm.distillation import *
from trainers.lm.ppo import *
from trainers.lm.grpo import *
from trainers.lm.agent import *
from trainers.lm.rollout_engine import *
from trainers.lm.train_tokenizer import *
from trainers.vlm.pretrain import *
from trainers.vlm.full_sft import *
from trainers.vam.full_sft import *