config.py

from transformers import PretrainedConfig


class LiteVit5Config(PretrainedConfig):
    """
    Configuration class for LiteVit5ForConditionalGeneration.
    """
    
    model_type = "litevit5"
    
    def __init__(
        self,
        pad_token_id: int = 0,
        eos_token_id: int = 1,
        decoder_start_token_id: int = 0,
        **kwargs
    ):
        super().__init__(
            pad_token_id=pad_token_id,
            eos_token_id=eos_token_id,
            decoder_start_token_id=decoder_start_token_id,
            **kwargs
        )