models-json/pretrained_models.json

{
    "uma-s-1": {
        "filename": "uma-s-1.pt",
        "repo_id": "facebook/UMA",
        "subfolder": "checkpoints",
        "atom_refs": {
            "subfolder": "references",
            "filename": "iso_atom_elem_refs.yaml"
        }
    },
    "uma-s-1p1": {
        "filename": "uma-s-1p1.pt",
        "repo_id": "facebook/UMA",
        "subfolder": "checkpoints",
        "atom_refs": {
            "subfolder": "references",
            "filename": "iso_atom_elem_refs.yaml"
        }
    },
    "uma-m-1p1": {
        "filename": "uma-m-1p1.pt",
        "repo_id": "facebook/UMA",
        "subfolder": "checkpoints",
        "atom_refs": {
            "subfolder": "references",
            "filename": "iso_atom_elem_refs.yaml"
        }
    },
    "esen-md-direct-all-omol": {
        "filename": "esen_md_direct_all.pt",
        "repo_id": "facebook/OMol25",
        "subfolder": "checkpoints",
        "atom_refs": {
            "subfolder": "references",
            "filename": "iso_atom_elem_refs.yaml"
        }
    },
    "esen-sm-conserving-all-omol": {
        "filename": "esen_sm_conserving_all.pt",
        "repo_id": "facebook/OMol25",
        "subfolder": "checkpoints",
        "atom_refs": {
            "subfolder": "references",
            "filename": "iso_atom_elem_refs.yaml"
        }
    },
    "esen-sm-direct-all-omol": {
        "filename": "esen_sm_direct_all.pt",
        "repo_id": "facebook/OMol25",
        "subfolder": "checkpoints",
        "atom_refs": {
            "subfolder": "references",
            "filename": "iso_atom_elem_refs.yaml"
        }
    }
}