Document released inference weights and download workflow
Browse files
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.
|