Skip to content

Commit 4384816

Browse files
authored
fix mistakes change in tp (#2550)
1 parent c861b78 commit 4384816

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddleformers/transformers/conversion_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1278,7 +1278,7 @@ def get_tensor_parallel_convert_actions(
12781278

12791279
@classmethod
12801280
def convert_tensor_parallel(
1281-
cls, model, weight_file: str, config: PretrainedConfig, state_dict=None, ignore_error=False
1281+
cls, weight_file: str, config: PretrainedConfig, state_dict=None, ignore_error=False
12821282
) -> None:
12831283
"""the entry of converting config and converting model file
12841284
@@ -1287,7 +1287,7 @@ def convert_tensor_parallel(
12871287
config (PretrainedConfig): the PretrainedConfig instance of model
12881288
"""
12891289

1290-
name_action_mappings = cls._get_tensor_parallel_mappings(model, config)
1290+
name_action_mappings = cls._get_tensor_parallel_mappings(config)
12911291
if config.quantization_config.is_weight_quantize():
12921292
name_action_mappings = add_quant_mapping(name_action_mappings, config.quantization_config)
12931293
if state_dict is None:

0 commit comments

Comments
 (0)