from transformers import PretrainedConfig


class NanbeigeVLMConfig(PretrainedConfig):
    """
    Configuration for NanbeigeVLM.

    Args:
        vision_model_id: HuggingFace model ID for the vision tower.
        llm_model_id:    HuggingFace model ID for the language model.
    """

    model_type = "nanbeige_vlm"

    def __init__(
        self,
        vision_model_id: str = "google/siglip-so400m-patch14-384",
        llm_model_id: str = "Nanbeige/Nanbeige4.1-3B",
        **kwargs,
    ):
        super().__init__(**kwargs)
        self.vision_model_id = vision_model_id
        self.llm_model_id = llm_model_id