configuration.py

from transformers import PretrainedConfig


class ExtSummConfig(PretrainedConfig):
    model_type = "pubmedbert-bio-ext-summ"

    def __init__(
        self, 
        input_size: int = 512,
        **kwargs
    ):
        self.input_size = input_size
        super().__init__(**kwargs)