from transformers import PretrainedConfig

class BBBConfig(PretrainedConfig):
  model_type = "bbb-model"

  def __init__(
    self,
    d_tab : int = 384, 
    d_img: int = 2048, 
    d_txt: int = 768, 
    proj_dim: int = 2048, 
    dropout: float = 0.1, 
    task: str = 'classification',
    **kwargs):

    self.d_tab = d_tab
    self.d_img = d_img
    self.d_txt = d_txt
    self.proj_dim = proj_dim
    self.dropout = dropout
    self.task = task
    super().__init__(**kwargs)