mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-04-12 23:22:42 +00:00
5 lines
219 B
Python
5 lines
219 B
Python
from colossalai.zero.sharded_param.sharded_tensor import ShardedTensor
|
|
from colossalai.zero.sharded_param.sharded_param import ShardedParam, ShardedParamV2
|
|
|
|
__all__ = ['ShardedParam', 'ShardedTensor', 'ShardedParamV2']
|