IEETA/BioNExt-Extractor

1 2from transformers import PretrainedConfig, AutoConfig3from typing import List4 5 6class BioNExtExtractorConfig(PretrainedConfig):7    model_type = "relation-novelty-extractor"8 9    def __init__(10        self,11        arch_type = "mha",12        index_type = "both",13        novel = True,14        tokenizer_special_tokens = ['[s1]','[e1]', '[s2]','[e2]' ],15        update_vocab = None,16        version="0.1.1",17        **kwargs,18    ):19        self.version = version20        self.arch_type = arch_type21        self.index_type = index_type22        self.novel = novel23        self.tokenizer_special_tokens = tokenizer_special_tokens24        self.update_vocab = update_vocab25        super().__init__(**kwargs)26        27 28