mirror of https://github.com/hpcaitech/ColossalAI
fix typo in constants (#1027)
parent
007ca0df92
commit
9b0c037027
|
@ -4,7 +4,7 @@
|
||||||
ALLOWED_MODES = [None, '1d', '2d', '2.5d', '3d', 'sequence']
|
ALLOWED_MODES = [None, '1d', '2d', '2.5d', '3d', 'sequence']
|
||||||
TENSOR_PARALLEL_MODE = 'tensor_parallel_mode'
|
TENSOR_PARALLEL_MODE = 'tensor_parallel_mode'
|
||||||
|
|
||||||
# intializer
|
# initializer
|
||||||
INITIALIZER_MAPPING = {
|
INITIALIZER_MAPPING = {
|
||||||
'data': 'Initializer_Data',
|
'data': 'Initializer_Data',
|
||||||
'tensor': 'Initializer_Tensor',
|
'tensor': 'Initializer_Tensor',
|
||||||
|
|
Loading…
Reference in New Issue