Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
JunnYu committed Dec 19, 2023
1 parent 259fa80 commit 1f808a9
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions paddlenlp/transformers/tokenizer_utils_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -1557,8 +1557,9 @@ def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
)
tokenizer_config_file_dir_list = set()
for k, v in resolved_vocab_files.items():
tokenizer_config_file_dir_list.add(os.path.dirname(v))
assert len(tokenizer_config_file_dir_list) <= 1, "All tokenizer files should be in the same directory."
if v is not None and os.path.isfile(v):
tokenizer_config_file_dir_list.add(os.path.dirname(v))
assert len(tokenizer_config_file_dir_list) == 1, "All tokenizer files should be in the same directory."
# Prepare tokenizer initialization kwargs
# Did we saved some inputs and kwargs to reload ?
has_tokenizer_file = resolved_vocab_files.get("tokenizer_file", None) is not None
Expand Down

0 comments on commit 1f808a9

Please sign in to comment.