configs/model.json

{
    "data": {
        "tokenizer_path": "tokenizer",
        "protein_h5": "PLACEHOLDER.h5",
        "max_generated_rna_bp": 101,
        "append_eos_token": true
    },
    "model": {
        "use_FiLM": false,
        "use_AdaLN": true,
        "use_gated_bias": false,
        "use_protein_conditioning_attention": true,
        "protein_dim": 2560,
        "num_hidden_layers": 12
    },
    "train": {
        "use_bf16": true,
        "t_eps": 0.001
    }
}