Update config.json

config.json

@@ -1 +1,12 @@
-{"feature_dim": 1024, "embedding_dim": 256, "hidden_dim": 512, "vocab_size": 5000, "num_layers": 1, "dropout": 0.5}
+{
+  "feature_dim": 1024,
+  "embedding_dim": 256,
+  "hidden_dim": 512,
+  "vocab_size": 5000,
+  "num_layers": 1,
+  "dropout": 0.5,
+  "architectures": ["modeling.ImageCaptioningModel"],
+  "auto_map": {
+    "AutoModel": "modeling.ImageCaptioningModel"
+  }
+}