Hugging Face模型权重下载，以ESM系列为例

雪垆

已于 2024-12-13 16:42:22 修改

阅读量716

点赞数 3

文章标签： python linux

于 2024-12-13 16:35:48 首次发布

本文链接：https://blog.csdn.net/qq_43611382/article/details/144453821

版权

huggingface模型权重pth文件下载，以ESM系列为例

无需登录的下载方式

ESM Cambrian（ESM C）的600M参数量的模型权重放在：
HuggingFace hub：https://huggingface.co/EvolutionaryScale/esmc-600m-2024-12
下载链接：https://huggingface.co/EvolutionaryScale/esmc-600m-2024-12/blob/main/data/weights/esmc_600m_2024_12_v0.pth
需要挂梯子访问，直接下载速度太慢。
可以从hf-mirror：https://hf-mirror.com/ 镜像网站下载：https://hf-mirror.com/EvolutionaryScale/esmc-600m-2024-12
下载链接：https://hf-mirror.com/EvolutionaryScale/esmc-600m-2024-12/blob/main/data/weights/esmc_600m_2024_12_v0.pth
可以通过直链存到迅雷云盘等云端，然后再下载。

需要登录的下载方式

ESM-3的esm3-sm-open-v1的模型权重放在：https://huggingface.co/EvolutionaryScale/esm3-sm-open-v1，需要登录账号申请访问，无法直接下载，且无法存到云盘。
只能通过 huggingface-cli工具从hf-mirror下载：

pip install -U huggingface_hub  -i https://pypi.tuna.tsinghua.edu.cn/simple/
export HF_ENDPOINT=https://hf-mirror.com
huggingface-cli download --token hf_xxxxxxxxxxxx --resume-download EvolutionaryScale/esm3-sm-open-v1 --local-dir data/weights

会将整个 EvolutionaryScale/esm3-sm-open-v1 项目所有文件下载下来。
其中需要 huggingface.co access token ，从 https://huggingface.co/settings/tokens 设置自己的token。