from transformers import Qwen3VLConfig


class OpsColQwen3Config(Qwen3VLConfig):
    """
    Configuration class for OpsColQwen3 model.
    """
    model_type = "ops_colqwen3"

    def __init__(
        self,
        dims: int = 2560,
        mask_non_image_embeddings: bool = False,
        **kwargs
    ):
        super().__init__(**kwargs)
        self.dims = dims
        self.mask_non_image_embeddings = mask_non_image_embeddings