config.json

{
  "feature_pooling_strategy": "SelfAttentionPooling",
  "finetune_seq_embed_model": false,
  "fps_num_heads": 4,
  "fps_return_weights": true,
  "fps_use_soft_pooling": true,
  "load_in_4bit": true,
  "lora_alpha": 16,
  "lora_dropout": 0.05,
  "lora_r": 32,
  "max_seq_len": 514,
  "seq_embed_model_chunk_size": 256,
  "seq_embed_model_name": "facebook/esm2_t33_650M_UR50D",
  "seq_feature_encoder": "GCNConv",
  "seq_feature_encoder_dropout": 0.2,
  "seq_feature_encoder_num_heads": 4,
  "stoichiometry_classes_to_use": [
    1,
    2,
    4,
    3,
    6,
    8,
    5,
    12,
    24,
    10,
    14,
    7,
    9,
    16
  ]
}