from transformers import PretrainedConfig


class RADARConfig(PretrainedConfig):
    """HuggingFace-compatible configuration for the RADAR model."""

    model_type = "radar"

    def __init__(
        self,
        base_model_name: str = "answerdotai/ModernBERT-large",
        style_dim: int = 38,
        style_proj_dim: int = 128,
        fusion_dim: int = 512,
        num_attention_heads: int = 8,
        dropout: float = 0.1,
        **kwargs,
    ):

        self.base_model_name = base_model_name
        self.style_dim = style_dim
        self.style_proj_dim = style_proj_dim
        self.fusion_dim = fusion_dim
        self.num_attention_heads = num_attention_heads
        self.dropout = dropout
        super().__init__(**kwargs)