Upload TypeBERTForSequenceClassification

config.json

@@ -23,6 +23,7 @@
     "virtual": 4
   },
   "model_type": "type_bert",
+  "tokenizer_class": "bert-base-uncased",
   "torch_dtype": "float32",
   "transformers_version": "4.22.1"
 }

type_bert_model.py

@@ -28,6 +28,7 @@ class TypeBERTConfig(PretrainedConfig):
         }
         
         self.architectures = ['TypeBERTForSequenceClassification']
+        self.tokenizer_class = 'bert-base-uncased'