AI4Protein/VenusFactory2-ckpts

VenusFactory2 Checkpoints

Pretrained / finetuned weights for VenusFactory2.

These files are not stored in the GitHub repository. VenusFactory2 downloads them into a local ckpt/ directory.

Install / download

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

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

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