# configuration_khmerocr.py
from transformers import PretrainedConfig

class KhmerOCRConfig(PretrainedConfig):
    model_type = "khmer-ocr"

    def __init__(
        self,
        vocab_size=124,      
        emb_dim=384,
        max_global_len=4096,
        pad_idx=0,
        nhead=8,
        num_encoder_layers=2,
        num_decoder_layers=2,
        **kwargs,
    ):
        self.vocab_size = vocab_size
        self.emb_dim = emb_dim
        self.max_global_len = max_global_len
        self.pad_idx = pad_idx
        self.nhead = nhead
        self.num_encoder_layers = num_encoder_layers
        self.num_decoder_layers = num_decoder_layers
        super().__init__(**kwargs)