from transformers import PretrainedConfig


class NanbeigeVLMConfig(PretrainedConfig):
    model_type = "nanbeige_vlm"

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