| { |
| "format": "la-ribo-inference-bundle", |
| "format_version": 1, |
| "code_repository": "https://github.com/GENTEL-lab/La-Ribo", |
| "paper": "https://arxiv.org/abs/2610.12236", |
| "checkpoint_format": "pytorch_state_dict", |
| "variants": { |
| "base": { |
| "checkpoint": "weights/la_ribo_base.pt", |
| "weights": "ema", |
| "inference_config": "config/la-ribo-base.yaml", |
| "flow_model": { |
| "latent_dim": 16, |
| "explicit_backbone_mode": "p_c4prime_n", |
| "coarse_dim": 9, |
| "relpos_dim": 64, |
| "dist_num_bins": 16, |
| "dist_min": 0.0, |
| "dist_max": 32.0, |
| "time_emb_dim": 64, |
| "token_dim": 512, |
| "pair_dim": 256, |
| "cond_dim": 128, |
| "num_layers": 12, |
| "num_heads": 8, |
| "use_qkln": true, |
| "transition_expansion": 4, |
| "pair_update": { |
| "enabled": false |
| } |
| }, |
| "autoencoder": { |
| "num_nt_types": 4, |
| "num_torsions": 10, |
| "latent_dim": 16, |
| "explicit_backbone": { |
| "mode": "p_c4prime_n", |
| "zero_center": true |
| }, |
| "decoder_conf": { |
| "relpos_dim": 64, |
| "dist_num_bins": 16, |
| "dist_min": 0.0, |
| "dist_max": 32.0, |
| "token_dim": 256, |
| "pair_dim": 128, |
| "cond_dim": 64, |
| "num_layers": 6, |
| "num_heads": 8, |
| "use_qkln": true |
| } |
| } |
| }, |
| "tri": { |
| "checkpoint": "weights/la_ribo_tri.pt", |
| "weights": "ema", |
| "inference_config": "config/la-ribo-tri.yaml", |
| "flow_model": { |
| "latent_dim": 16, |
| "explicit_backbone_mode": "p_c4prime_n", |
| "coarse_dim": 9, |
| "relpos_dim": 64, |
| "dist_num_bins": 16, |
| "dist_min": 0.0, |
| "dist_max": 32.0, |
| "time_emb_dim": 64, |
| "token_dim": 512, |
| "pair_dim": 256, |
| "cond_dim": 128, |
| "num_layers": 12, |
| "num_heads": 8, |
| "use_qkln": true, |
| "transition_expansion": 4, |
| "pair_update": { |
| "enabled": true, |
| "every": 2, |
| "use_cuequivariance": true, |
| "expansion": 2, |
| "token_to_pair": "add" |
| } |
| }, |
| "autoencoder": { |
| "num_nt_types": 4, |
| "num_torsions": 10, |
| "latent_dim": 16, |
| "explicit_backbone": { |
| "mode": "p_c4prime_n", |
| "zero_center": true |
| }, |
| "decoder_conf": { |
| "relpos_dim": 64, |
| "dist_num_bins": 16, |
| "dist_min": 0.0, |
| "dist_max": 32.0, |
| "token_dim": 256, |
| "pair_dim": 128, |
| "cond_dim": 64, |
| "num_layers": 6, |
| "num_heads": 8, |
| "use_qkln": true |
| } |
| } |
| } |
| }, |
| "shared_files": { |
| "autoencoder": "weights/la_ribo_vae.pt", |
| "latent_stats": "weights/latent_stats.pt", |
| "coarse_stats": "weights/coarse_stats.pt" |
| }, |
| "sampling": { |
| "solver": "forward_euler", |
| "num_timesteps": 200, |
| "scaffold_power": 1.25, |
| "latent_power": 2.0, |
| "self_condition": true |
| }, |
| "supported_devices": [ |
| "cpu", |
| "cuda" |
| ], |
| "benchmark_lengths": [ |
| 40, |
| 50, |
| 60, |
| 70, |
| 80, |
| 90, |
| 100, |
| 110, |
| 120, |
| 130, |
| 140, |
| 150 |
| ] |
| } |
|
|
