| 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, | |
| version="0.1.0", | |
| **kwargs, | |
| ): | |
| self.version = version | |
| self.arch_type = arch_type | |
| self.index_type = index_type | |
| self.novel = novel | |
| super().__init__(**kwargs) | |