File tree Expand file tree Collapse file tree 1 file changed +11
-5
lines changed
vllm/transformers_utils/tokenizers Expand file tree Collapse file tree 1 file changed +11
-5
lines changed Original file line number Diff line number Diff line change @@ -122,15 +122,21 @@ def list_local_repo_files(repo_id: str, revision: Optional[str]) -> list[str]:
122
122
123
123
124
124
def find_tokenizer_file (files : list [str ]):
125
+ # Accept both versioned (tokenizer.model.v3) and unversioned
126
+ # (tokenizer.model) forms, plus tekken.json and tokenizer.mm.model
127
+ # variants. Previous pattern only matched the versioned variants.
125
128
file_pattern = re .compile (
126
- r"^tokenizer\.model\.v.*$|^tekken\.json$|^tokenizer\.mm\.model\.v.*$" )
129
+ r"^tokenizer\.model(\.v.*)?|tekken\.json|tokenizer\.mm\.model(\.v.*)?$"
130
+ )
127
131
128
132
matched_files = [file for file in files if file_pattern .match (file )]
129
133
if len (matched_files ) > 1 :
130
- raise OSError (
131
- f"Found { len (matched_files )} files matching the "
132
- f"pattern: `{ file_pattern .pattern } `. Make sure only one Mistral "
133
- f"tokenizer is present in { files } ." )
134
+ logger .warning (
135
+ "Multiple files matched pattern `%s`: %s. Using %s." ,
136
+ file_pattern .pattern ,
137
+ matched_files ,
138
+ matched_files [0 ],
139
+ )
134
140
elif len (matched_files ) == 0 :
135
141
raise OSError (
136
142
f"Found { len (matched_files )} files matching the "
You can’t perform that action at this time.
0 commit comments