"""Config for TinyBrainBot recurrent-depth models (fields mirror the native TinyBrainBotConfig)."""
from transformers import PretrainedConfig


class TBBLoopConfig(PretrainedConfig):
    model_type = "tbbloop"

    def __init__(self, native=None, eval_loops=0, tie_word_embeddings=True, **kwargs):
        self.native = native or {}
        self.eval_loops = eval_loops
        super().__init__(tie_word_embeddings=tie_word_embeddings, **kwargs)