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

By default unimol_tools first tries the official Hugging Face endpoint. If that fails and HF_ENDPOINT is not set, it automatically retries with https://hf-mirror.com. Set the variable yourself to choose a specific endpoint.

unimol_tools.weights.weight_hub.py control the logger.