Gunulhona
/

tb_pretrained_sts

Feature Extraction

Inference Endpoints

Model card Files Files and versions Community

Gunulhona commited on Nov 29, 2023

Commit

230f0db

•

1 Parent(s): cda102b

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -11,13 +11,13 @@
   "classifier_dropout": 0.0,
   "d_model": 768,
   "decoder_attention_heads": 12,
-  "decoder_ffn_dim": 4096,
   "decoder_layerdrop": 0.0,
   "decoder_layers": 9,
   "decoder_start_token_id": 2,
   "dropout": 0.1,
   "encoder_attention_heads": 12,
-  "encoder_ffn_dim": 4096,
   "encoder_layerdrop": 0.0,
   "encoder_layers": 3,
   "eos_token_id": 2,

   "classifier_dropout": 0.0,
   "d_model": 768,
   "decoder_attention_heads": 12,
+  "decoder_ffn_dim": 3072,
   "decoder_layerdrop": 0.0,
   "decoder_layers": 9,
   "decoder_start_token_id": 2,
   "dropout": 0.1,
   "encoder_attention_heads": 12,
+  "encoder_ffn_dim": 3072,
   "encoder_layerdrop": 0.0,
   "encoder_layers": 3,
   "eos_token_id": 2,