File size: 463 Bytes
086d355 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from transformers import PretrainedConfig
class SentenceEmbedderConfig(PretrainedConfig):
model_type = 'sentence_embedder'
def __init__(
self,
backbone_name: str = 'xlm-roberta-base',
output_size: int = 768,
init_backbone: bool = False,
**kwargs
):
self.backbone_name = backbone_name
self.output_size = output_size
self.init_backbone = init_backbone
super().__init__(**kwargs) |