Skip to content

Commit 82a7177

Browse files
authored
quick fix os.path.split (#8508)
1 parent bbf945b commit 82a7177

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddlenlp/transformers/model_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1916,7 +1916,7 @@ def _fuse_or_split_keys(
19161916
if (
19171917
shard_file.endswith(".safetensors")
19181918
and config.tensor_parallel_degree > 1
1919-
and "tp" not in os.path.spilt(shard_file)[-1]
1919+
and "tp" not in os.path.split(shard_file)[-1]
19201920
):
19211921
pre_tensor_parallel_split = True
19221922
assert loaded_keys is not None, "loaded_keys is not None."

0 commit comments

Comments
 (0)