from transformers import PretrainedConfig


class PredNetConfig(PretrainedConfig):
    model_type = "prednet"

    def __init__(self, sequence_len=4, resize_hw=(128, 128), **kwargs):
        super().__init__(**kwargs)
        self.sequence_len = sequence_len
        self.resize_hw = list(resize_hw)