File size: 1,811 Bytes
46494b4 | 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 | ---
license: mit
tags:
- biology
- protein
- venusfactory
- checkpoints
library_name: venus-factory
---
# VenusFactory2 Checkpoints
Pretrained / finetuned weights for [VenusFactory2](https://github.com/AI4Protein/VenusFactory2).
These files are **not** stored in the GitHub repository. VenusFactory2 downloads them into a local `ckpt/` directory.
## Install / download
```bash
# From a VenusFactory2 checkout
python scripts/download_ckpts.py --preset predict-core # demo + ankh-large adapters
python scripts/download_ckpts.py --preset proteinmpnn # ProteinMPNN
python scripts/download_ckpts.py --preset all # everything
python scripts/download_ckpts.py --include 'DeepSol/**' # one task
```
Or with the Hugging Face CLI:
```bash
huggingface-cli download AI4Protein/VenusFactory2-ckpts \
--local-dir ckpt \
--include "DeepSol/**" "demo/**"
```
## Auto-download
VenusFactory2 enables on-demand fetch by default (`VENUS_CKPT_AUTO_DOWNLOAD=1`).
The first time a tool needs a missing adapter, it pulls from this repo into `ckpt/`.
| Variable | Default | Meaning |
|----------|---------|---------|
| `VENUS_CKPT_REPO_ID` | `AI4Protein/VenusFactory2-ckpts` | This repo |
| `VENUS_CKPT_DIR` | `ckpt` | Local cache root |
| `VENUS_CKPT_REVISION` | `main` | Branch / tag / commit |
| `VENUS_CKPT_AUTO_DOWNLOAD` | `1` | Fetch missing weights automatically |
## Layout
```text
demo/
DeepSol/<backbone>/*.pt|*.json
ProteinMPNN/vanilla_model_weights/*.pt
...
manifest.json
```
See `manifest.json` for the full file list, sizes, and sha256 checksums.
## Presets
- `demo` — demo solubility adapter
- `predict-core` — demo + all task `ankh-large` adapters
- `predict-all` — demo + all finetuned task folders
- `proteinmpnn` — ProteinMPNN weights
- `all` — entire repository
|