mrzz commited on
Commit
9ba63c4
·
verified ·
1 Parent(s): f6e97ee

Document released inference weights and download workflow

Browse files
Files changed (1) hide show
  1. README.md +32 -0
README.md CHANGED
@@ -4,6 +4,38 @@ license: mit
4
 
5
  # La-Ribo
6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  ## Data
8
 
9
  The `data/` directory contains 10,631 RNA sequences together with 3D structures predicted by trRosettaRNA2 and per-residue pLDDT scores.
 
4
 
5
  # La-Ribo
6
 
7
+ ## Pretrained weights
8
+
9
+ The released inference bundle is available in [weights/](https://huggingface.co/GENTEL-Lab/La-Ribo/tree/main/weights).
10
+
11
+ | File | Purpose |
12
+ |---|---|
13
+ | `weights/la_ribo_base.pt` | Base flow model, exported EMA weights |
14
+ | `weights/la_ribo_tri.pt` | Tri flow model, exported EMA weights |
15
+ | `weights/la_ribo_vae.pt` | Shared autoencoder weights |
16
+ | `weights/latent_stats.pt` | Latent whitening statistics |
17
+ | `weights/coarse_stats.pt` | Scaffold scaling statistics |
18
+
19
+ The three model files are pure tensor state dictionaries. The normalization files contain only inference-required values and format fields. Base and Tri already contain EMA weights; no `--use_ema` flag is required.
20
+
21
+ ### Download the inference bundle
22
+
23
+ Run this from the local [La-Ribo inference repository](https://github.com/GENTEL-lab/La-Ribo):
24
+
25
+ ```bash
26
+ uvx --from huggingface_hub hf download GENTEL-Lab/La-Ribo \
27
+ weights/la_ribo_base.pt \
28
+ weights/la_ribo_tri.pt \
29
+ weights/la_ribo_vae.pt \
30
+ weights/latent_stats.pt \
31
+ weights/coarse_stats.pt \
32
+ --local-dir .
33
+ ```
34
+
35
+ Both variants use the shared autoencoder and both statistics files. Choose `config/la-ribo-base.yaml` with `la_ribo_base.pt`, or `config/la-ribo-tri.yaml` with `la_ribo_tri.pt`. Installation and CPU/CUDA inference commands are in the [GitHub README](https://github.com/GENTEL-lab/La-Ribo#readme). The default sampling protocol uses 200 steps and scaffold/latent powers 1.25/2.0.
36
+
37
+ The data below are optional for inference.
38
+
39
  ## Data
40
 
41
  The `data/` directory contains 10,631 RNA sequences together with 3D structures predicted by trRosettaRNA2 and per-residue pLDDT scores.