KhaldiAbderrhmane
/

bert-multihead-attention-similarity

Feature Extraction

bert_multi_attention

Model card Files Files and versions Community

KhaldiAbderrhmane commited on Jun 6, 2024

Commit

c128140

·

verified ·

1 Parent(s): 481c393

Upload model

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -4,7 +4,7 @@
   ],
   "dropout": 0.1,
   "hidden_size": 768,
-  "model_type": "bert-multihead-attention-similarity",
   "num_heads": 8,
   "rnn_bidirectional": true,
   "rnn_hidden_size": 128,

   ],
   "dropout": 0.1,
   "hidden_size": 768,
+  "model_type": "bert_multi_attention",
   "num_heads": 8,
   "rnn_bidirectional": true,
   "rnn_hidden_size": 128,