File size: 722 Bytes
f4c1f6e 1e84124 f4c1f6e 1e84124 f4c1f6e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
from transformers import PretrainedConfig, AutoConfig
from typing import List
class BioNExtExtractorConfig(PretrainedConfig):
model_type = "relation-novelty-extractor"
def __init__(
self,
arch_type = "mha",
index_type = "both",
novel = True,
tokenizer_special_tokens = ['[s1]','[e1]', '[s2]','[e2]' ],
update_vocab = None,
version="0.1.1",
**kwargs,
):
self.version = version
self.arch_type = arch_type
self.index_type = index_type
self.novel = novel
self.tokenizer_special_tokens = tokenizer_special_tokens
self.update_vocab = update_vocab
super().__init__(**kwargs)
|