"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "34ccc8ebf4c9e440e4621dab6cdb0a8f2dc903ad"
Unverified Commit eb3479e7 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Some models have no tokenizers (#11064)

parent 773e4c72
...@@ -69,13 +69,14 @@ def merge_model_tokenizer_mappings( ...@@ -69,13 +69,14 @@ def merge_model_tokenizer_mappings(
model_tokenizer_mapping = OrderedDict([]) model_tokenizer_mapping = OrderedDict([])
for configuration in configurations: for configuration in configurations:
model = model_mapping[configuration] if configuration in model_mapping and configuration in tokenizer_mapping:
tokenizer = tokenizer_mapping[configuration][0] model = model_mapping[configuration]
tokenizer_fast = tokenizer_mapping[configuration][1] tokenizer = tokenizer_mapping[configuration][0]
tokenizer_fast = tokenizer_mapping[configuration][1]
model_tokenizer_mapping.update({tokenizer: (configuration, model)})
if tokenizer_fast is not None: model_tokenizer_mapping.update({tokenizer: (configuration, model)})
model_tokenizer_mapping.update({tokenizer_fast: (configuration, model)}) if tokenizer_fast is not None:
model_tokenizer_mapping.update({tokenizer_fast: (configuration, model)})
return model_tokenizer_mapping return model_tokenizer_mapping
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment