---
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