From 06c85144c653ab6d5ad7baecb23033b9dc3f5a52 Mon Sep 17 00:00:00 2001 From: DesmonDay <908660116@qq.com> Date: Wed, 29 May 2024 16:49:11 +0800 Subject: [PATCH] quick fix os.path.split --- paddlenlp/transformers/model_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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."