| 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 | |
| ) | |