File size: 519 Bytes
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 |
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)
|