from transformers import PretrainedConfig

class ChronoGridConfig(PretrainedConfig):
    model_type = "chronogrid"

    def __init__(
        self,
        img_size=227,
        num_classes=11,
        **kwargs
    ):
        self.img_size = img_size
        self.num_classes = num_classes
        super().__init__(**kwargs)