github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

Co-authored-by: ric <mkkt_bkkt@mail.ustc.edu.cn>
pull/1743/head
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']