diff --git a/tokenizer_config.json b/tokenizer_config.json index 8360f55..84fe21c 100644 --- a/tokenizer_config.json +++ b/tokenizer_config.json @@ -3,6 +3,10 @@ "do_basic_tokenize": true, "do_lower_case": true, "mask_token": "[MASK]", + "model_input_names": [ + "input_ids", + "attention_mask" + ], "model_max_length": 512, "name_or_path": "ybelkada/blip-image-captioning-base", "never_split": null,