ColossalAI/colossalai/shardformer/__init__.py

2 lines
44 B
Python

from .shard import ShardConfig, shard_model