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

4 lines
83 B
Python

from .sharded_optim_v2 import ShardedOptimizerV2
__all__ = ["ShardedOptimizerV2"]