__init__.py — VeOmni 引擎模块初始化

文件概述

导出 VeOmniEngine 和 VeOmniEngineWithLMHead。

from .transformer_impl import VeOmniEngine, VeOmniEngineWithLMHead

小结

VeOmni 引擎模块的入口，VeOmni 是字节跳动的分布式训练框架。