ProRiboGen / generator /config_sample.json
TimelessAEther's picture
Upload ProRiboGen inference package and checkpoints
6dd9839 verified
Raw
History Blame Contribute Delete
839 Bytes
{
"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
},
"sample": {
"checkpoint": "PLACEHOLDER.pt",
"output_dir": "PLACEHOLDER_OUT",
"num_sequences_per_rbp": 256,
"length_bp": 80,
"min_rna_bp": 12,
"steps": 50,
"temperature": 1.0,
"top_p": 0.90,
"top_k": 0,
"alg": "maskgit_plus",
"alg_temp": 0.8,
"variable_length": true
}
}