ProRiboGen / configs /paths.env.example
TimelessAEther's picture
Upload ProRiboGen inference package and checkpoints
6dd9839 verified
Raw
History Blame Contribute Delete
841 Bytes
# Copy to paths.env and edit for your GPU server.
# Do not commit machine-specific absolute paths to a public repo.
# ===== VESM (required; not shipped in this package) =====
# Option A — standard 17-VESM3 root:
VESM_ROOT=/path/to/17-VESM3
# expects:
# $VESM_ROOT/models/base/facebook_esm2_t36_3B_UR50D
# $VESM_ROOT/models/weights/VESM_3B.pth
# Option B — explicit paths (overrides A if both set in the loader):
# VESM_BASE_MODEL_DIR=/path/to/facebook_esm2_t36_3B_UR50D
# VESM_WEIGHTS=/path/to/VESM_3B.pth
# ===== Package-relative paths (usually leave as-is) =====
GENERATOR_CKPT=checkpoints/generator.pt
BACKBONE_CKPT=checkpoints/backbone.pt
CLASSIFIER_CKPT=checkpoints/classifier.pt
SAMPLE_CONFIG=configs/sample_fixed80bp.json
MODEL_CONFIG=configs/model.json
TOKENIZER_DIR=generator/tokenizer
WORKSPACE=workspace
DEVICE=cuda:0