fix vocab and config

README.md

@@ -4,7 +4,7 @@ It was created as follows in a python shell
 
 ```python
 import transformers
-config = transformers.DistilBertConfig(vocab_size=1, n_layers=1, n_heads=1, dim=1, hidden_dim=4 * 1, num_labels=2, id2label={0: "negative", 1: "positive"}, label2id={"negative": 0, "positive": 1})
+config = transformers.DistilBertConfig(vocab_size=5, n_layers=1, n_heads=1, dim=1, hidden_dim=4 * 1, num_labels=2, id2label={0: "negative", 1: "positive"}, label2id={"negative": 0, "positive": 1})
 model = transformers.DistilBertForSequenceClassification(config)
 tokenizer = transformers.DistilBertTokenizer("/tmp/empty_vocab.txt", model_max_length=512)
 config.save_pretrained(".")

config.json

@@ -1,5 +1,8 @@
 {
   "activation": "gelu",
+  "architectures": [
+    "DistilBertForSequenceClassification"
+  ],
   "attention_dropout": 0.1,
   "dim": 1,
   "dropout": 0.1,
@@ -22,5 +25,5 @@
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,
   "transformers_version": "4.4.2",
-  "vocab_size": 1
+  "vocab_size": 5
 }

vocab.txt

@@ -0,0 +1,5 @@
+[UNK]
+[SEP]
+[PAD]
+[CLS]
+[MASK]