File size: 553 Bytes
244d6df |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
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
)
|