diff --git a/paddlenlp/transformers/model_utils.py b/paddlenlp/transformers/model_utils.py index 722bde20ee70..f56ad381f36e 100644 --- a/paddlenlp/transformers/model_utils.py +++ b/paddlenlp/transformers/model_utils.py @@ -1916,7 +1916,7 @@ def _fuse_or_split_keys( if ( shard_file.endswith(".safetensors") and config.tensor_parallel_degree > 1 - and "tp" not in os.path.spilt(shard_file)[-1] + and "tp" not in os.path.split(shard_file)[-1] ): pre_tensor_parallel_split = True assert loaded_keys is not None, "loaded_keys is not None."