from transformers import PretrainedConfig


class TimesformerScrollprizeConfig(PretrainedConfig):
    # model_type = "timesformerscrollprize"

    def __init__(
        self,
        window_size=64,
        depth=8,
        n_heads=6,
        patch_size=16,
        num_frames=26,
        num_classes=16,
        dim=512,
        **kwargs,
    ):
        self.window_size=window_size
        self.depth=depth
        self.n_heads=n_heads
        self.patch_size=patch_size
        self.num_frames=num_frames
        self.num_classes=num_classes
        self.dim=dim
        super().__init__(**kwargs)