ColossalAI/colossalai/zero/sharded_model/__init__.py
2022-03-25 14:54:39 +08:00

3 lines
74 B
Python

from .sharded_model_v2 import ShardedModelV2
__all__ = ['ShardedModelV2']