Update config.json

config.json

@@ -1,8 +1,18 @@
 {
-  "_name_or_path": "/content/hierarchical-transformers/data/PLMs/hi-transformer-p1-grouped/",
+  "_name_or_path": "igorktech/hierarchical-dialog-bert",
   "architectures": [
     "HATForMaskedLM"
   ],
+   "auto_map": {
+     "AutoConfig": "configuration_hat.HATConfig",
+     "AutoTokenizer": "tokenization_hat.HATTokenizer",
+     "AutoModel": "modelling_hat.HATModel",
+     "AutoModelForMaskedLM": "modelling_hat.HATForMaskedLM",
+     "AutoModelForMultipleChoice": "modelling_hat.HATForMultipleChoice",
+     "AutoModelForQuestionAnswering": "modelling_hat.HATForQuestionAnswering",
+     "AutoModelForSequenceClassification": "modelling_hat.HATForSequenceClassification",
+     "AutoModelForTokenClassification": "modelling_hat.HATForTokenClassification"
+   },
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,
   "encoder_layout": {