InternLM/internlm/moe/__init__.py

4 lines
82 B
Python

from internlm.moe.sharded_moe import GShardMOELayer
__all__ = ["GShardMOELayer"]