Weights

We recommend installing huggingface_hub so that the required Uni-Mol models can be automatically downloaded at runtime! It can be installed by:

pip install huggingface_hub

huggingface_hub allows you to easily download and manage models from the Hugging Face Hub, which is key for using Uni-Mol models.

Models in Huggingface

The Uni-Mol pretrained models can be found at dptech/Uni-Mol-Models.

If the download is slow, you can use other mirrors, such as:

export HF_ENDPOINT=https://hf-mirror.com

Setting the HF_ENDPOINT environment variable specifies the mirror address for the Hugging Face Hub to use when downloading models.

unimol_tools.weights.weight_hub.py control the logger.