INNER CODE UNIT · Python

install_models

PKU-EMBL/BASALT · BASALT_setup_China_mainland.py:459

def install_models(
    args: argparse.Namespace,
    executable: str,
    env_name: str,
    source: Path,
    environment: Dict[str, str],
) -> None:
    """Invoke the canonical model downloader inside the BASALT environment."""
    if args.model_source == "none":
        print("Skipping model weights (--model-source none).")
        return

    model_dir = Path(args.model_dir).expanduser().resolve()
    downloader = source / "BASALT" / "BASALT_models_download.py"
    command = [
        "python",
        str(downloader),
        "--path",

View source record →