quick fix os.path.split (PaddlePaddle#8508)

bo-ke · May 29, 2024 · 82a7177 · 82a7177
1 parent bbf945b
commit 82a7177
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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."