mirror of https://github.com/hpcaitech/ColossalAI
4 lines
130 B
Python
4 lines
130 B
Python
|
from .sharded_model import ShardedModel
|
||
|
|
from .sharded_model_v2 import ShardedModelV2
|
||
|
|
|
||
|
|
__all__ = ['ShardedModel', 'ShardedModelV2']
|