ZinengTang
/

tvlt-base

Inference Endpoints

Model card Files Files and versions Community

Zineng Tang commited on Dec 28, 2022

Commit

54f3160

•

1 Parent(s): dde2bf2

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -11,7 +11,7 @@
   "audio_size": 2048,
   "decoder_hidden_size": 512,
   "decoder_intermediate_size": 2048,
-  "decoder_num_attention_heads": 8,
   "decoder_num_hidden_layers": 8,
   "feature_size": 128,
   "hidden_act": "gelu",
@@ -25,7 +25,7 @@
   "label2id": {
     "LABEL_0": 0
   },
-  "layer_norm_eps": 1e-12,
   "model_type": "tvlt",
   "norm_pix_loss": true,
   "num_attention_heads": 12,

   "audio_size": 2048,
   "decoder_hidden_size": 512,
   "decoder_intermediate_size": 2048,
+  "decoder_num_attention_heads": 16,
   "decoder_num_hidden_layers": 8,
   "feature_size": 128,
   "hidden_act": "gelu",
   "label2id": {
     "LABEL_0": 0
   },
+  "layer_norm_eps": 1e-6,
   "model_type": "tvlt",
   "norm_pix_loss": true,
   "num_attention_heads": 12,