diff --git a/config.json b/config.json
index 9efe0e9..c50428f 100644
--- a/config.json
+++ b/config.json
@@ -1,8 +1,9 @@
 {
+  "_name_or_path": "./",
   "activation_dropout": 0.1,
   "apply_spec_augment": true,
   "architectures": [
-    "UniSpeechForPreTraining"
+    "UniSpeechForSequenceClassification"
   ],
   "attention_dropout": 0.1,
   "bos_token_id": 1,
@@ -46,8 +47,10 @@
   "layer_norm_eps": 1e-05,
   "layerdrop": 0.1,
   "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
   "mask_feature_prob": 0.0,
   "mask_time_length": 10,
+  "mask_time_min_masks": 2,
   "mask_time_prob": 0.05,
   "model_type": "unispeech",
   "num_attention_heads": 2,
@@ -63,7 +66,7 @@
   "proj_codevector_dim": 256,
   "replace_prob": 0.5,
   "torch_dtype": "float32",
-  "transformers_version": "4.12.0.dev0",
+  "transformers_version": "4.16.0.dev0",
   "use_weighted_layer_sum": false,
   "vocab_size": 32
 }
diff --git a/pytorch_model.bin b/pytorch_model.bin
index 9c9004b..54914fc 100644
--- a/pytorch_model.bin
+++ b/pytorch_model.bin
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c2643de148bfbb4889453163ff6aecd31f477e4a3afc71efcedf0d9f0c79021a
-size 841515
+oid sha256:d713e283c3fe7d58396361fa8f45320351951f4d3755fd1b3c45b15a480d1458
+size 159328