Hugging Face's logo

GENTEL-Lab
/
La-Ribo

Model card Files
xet
 Community
La-Ribo
File size: 3,236 Bytes
4b2384f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
{
  "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
  ]
}