mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-08 03:24:07 +00:00
[NFC] polish colossalai/zero/sharded_param/__init__.py code style (#1717)
Co-authored-by: ric <mkkt_bkkt@mail.ustc.edu.cn>
This commit is contained in:
parent
2b49ca80a3
commit
ea961d8fd1
@ -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']
|
||||
|
Loading…
Reference in New Issue
Block a user