SaeedLab/ProtEnrich-ProGen2

1from transformers import PretrainedConfig2 3class ProtEnrichConfig(PretrainedConfig):4  model_type = "protenrich"5 6  def __init__(7    self,8    seq_dim : int = 1536, 9    struct_dim: int = 1024, 10    dyn_dim: int = 20, 11    embed_dim: int = 1024, 12    project_dim: int = 256, 13    alpha_max: float = 0.3,14    num_labels: int = 2,15    **kwargs):16    super().__init__(**kwargs)17    self.seq_dim = seq_dim18    self.struct_dim = struct_dim19    self.dyn_dim = dyn_dim20    self.embed_dim = embed_dim21    self.project_dim = project_dim22    self.alpha_max = alpha_max23    self.num_labels = num_labels