mirror of https://github.com/hpcaitech/ColossalAI
5 lines
131 B
Python
5 lines
131 B
Python
|
from .sharded_param import ShardedParamV2
|
||
|
|
from .sharded_tensor import ShardedTensor
|
||
|
|
|
||
|
|
__all__ = ['ShardedTensor', 'ShardedParamV2']
|