space_utils/download_weights.py

import subprocess
import os

def download_ligandmpnn_weights():
    url = "https://files.ipd.uw.edu/pub/ligandmpnn/ligandmpnn_v_32_030_25.pt"
    command = f"wget {url} -O ./model_params/ligandmpnn_v_32_030_25.pt"
    os.makedirs("./model_params", exist_ok=True)
    subprocess.run(command, shell=True, check=True)
    return 0