studio-ousia
/

luke-base

named entity recognition

relation classification

question answering

Model card Files Files and versions

ikuyamada commited on Apr 11, 2021

Commit

f936ea9

·

1 Parent(s): 4649112

update config

Files changed (1) hide show

config.json +3 -2

config.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "LukeEntityAwareAttentionModel"
   ],
   "attention_probs_dropout_prob": 0.1,
   "bert_model_name": "roberta-base",
@@ -22,8 +22,9 @@
   "output_past": true,
   "pad_token_id": 1,
   "position_embedding_type": "absolute",
-  "transformers_version": "4.4.0.dev0",
   "type_vocab_size": 1,
   "use_cache": true,
   "vocab_size": 50267
 }

 {
   "architectures": [
+    "LukeModel"
   ],
   "attention_probs_dropout_prob": 0.1,
   "bert_model_name": "roberta-base",
   "output_past": true,
   "pad_token_id": 1,
   "position_embedding_type": "absolute",
+  "transformers_version": "4.5.0.dev0",
   "type_vocab_size": 1,
   "use_cache": true,
+  "use_entity_aware_attention": true,
   "vocab_size": 50267
 }