SaeedLab/BindScreen-Finetuning-LIT_PCBA

1from transformers import PretrainedConfig2 3 4class SeqScreenConfig(PretrainedConfig):5  model_type = "seqscreen"6 7  def __init__(8    self,9    prot_dim: int = 2560,10    mol_dim: int = 768,11    proj_dim: int = 512,12    dropout: float = 0.1,13    esm2_model_name: str = "facebook/esm2_t36_3B_UR50D",14    lora_adapter_repo: str = None,15    **kwargs,16  ):17    super().__init__(**kwargs)18    self.prot_dim = prot_dim19    self.mol_dim = mol_dim20    self.proj_dim = proj_dim21    self.dropout = dropout22    self.esm2_model_name = esm2_model_name23    self.lora_adapter_repo = lora_adapter_repo