如何修改hugging face的模型默认下载地址

hit56笔记

已于 2024-11-15 11:32:01 修改

阅读量1.2w

点赞数 11

文章标签： python

于 2023-06-14 10:01:46 首次发布

本文链接：https://blog.youkuaiyun.com/zh515858237/article/details/131201712

版权

文章介绍了如何通过设置环境变量来改变huggingface模型在Linux中的默认下载位置，以解决模型数据占用空间大的问题。提供了两种方法，一是使用命令行设置环境变量，二是直接在代码中指定路径，包括设置HF_DATASETS_CACHE,HF_HOME,HUGGINGFACE_HUB_CACHE和TRANSFORMERS_CACHE等变量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

huggingface的默认下载路径在~/.cache/huggingface/hub/，但模型数据占用空间往往很大，可以用以下方法修改默认下载路径。

方法一：在linux中指定环境变量

export HF_DATASETS_CACHE="/disk/cache/"
export HF_HOME="/disk/cache/"
export HUGGINGFACE_HUB_CACHE="/disk/cache/"
export TRANSFORMERS_CACHE="/disk/cache/"

方法二：在代码中指定环境变量

import os
os.environ["HF_DATASETS_CACHE"] = "/disk/cache/"
os.environ["HF_HOME"] = "/disk/cache/"
os.environ["HUGGINGFACE_HUB_CACHE"] = "/disk/cache/"
os.environ["TRANSFORMERS_CACHE"] = "/disk/cache/"