ColossalAI/colossalai/shardformer/shard/__init__.py

mirror of https://github.com/hpcaitech/ColossalAI
from .shard_config import ShardConfig
from .sharder import ModelSharder, shard_model
from .slicer import Slicer

__all__ = ['ShardConfig', 'ModelSharder', 'shard_model', 'Slicer']
[shardformer] refactored the user api (#3828) * [shardformer] refactored the user api * polish code 2023-05-24 08:01:26 +00:00			`from .shard_config import ShardConfig`
			`from .sharder import ModelSharder, shard_model`
			`from .slicer import Slicer`

			`__all__ = ['ShardConfig', 'ModelSharder', 'shard_model', 'Slicer']`