ColossalAI/colossalai/shardformer/shard/__init__.py

6 lines
180 B
Python

from .shard_config import ShardConfig
from .sharder import ModelSharder, shard_model
from .slicer import Slicer
__all__ = ['ShardConfig', 'ModelSharder', 'shard_model', 'Slicer']