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,