configuration_guardito.py

from transformers import PretrainedConfig

class GuarditoConfig(PretrainedConfig):
    model_type = "guardito"

    def __init__(
        self,
        base_model_name="deepvk/RuModernBERT-base",
        pooling_type="mean",
        dropout=0.1,
        threshold=0.5, # Дефолтный порог
        **kwargs,
    ):
        self.base_model_name = base_model_name
        self.pooling_type = pooling_type
        self.dropout = dropout
        self.threshold = threshold
        super().__init__(**kwargs)