| 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) |