ColossalAI/colossalai/legacy/zero/sharded_model/__init__.py

4 lines
75 B
Python

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