github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-10 12:22:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NFC] polish colossalai/zero/sharded_param/__init__.py code style (#1717)

Browse Source 
Co-authored-by: ric <mkkt_bkkt@mail.ustc.edu.cn>
This commit is contained in:
CsRic 2022-10-17 17:52:05 +08:00 committed by Frank Lee
parent 2b49ca80a3
commit ea961d8fd1
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
colossalai/zero/sharded_param/__init__.py
View File

@ -1,5 +1,4 @@
from colossalai.zero.sharded_param.sharded_tensor import ShardedTensor
from colossalai.zero.sharded_param.sharded_param import ShardedParamV2
__all__ = [
'ShardedTensor', 'ShardedParamV2']
__all__ = ['ShardedTensor', 'ShardedParamV2']