修改huggingface 的cache缓存路径

使用hugging face 来跑代码时,往往会把模型缓存到系统目录下,比如我的ubuntu系统就缓存到~.cache目录下。由于自己的home空间不足,先需要修改缓存的 .cache目录,修改方法如下:

找到自己安装的huggingface_hub下的constants.py 文件,比如我的文件位置在 lib/python3.10/site-packages/huggingface_hub 下。然后修改constants.py代码:

python 复制代码
# default cache
default_home = os.path.join(os.path.expanduser("~"), ".cache")
HF_HOME = os.path.expanduser(
    os.getenv(
        "HF_HOME",
        os.path.join(os.getenv("XDG_CACHE_HOME", default_home), "huggingface"),
    )
)
hf_cache_home = HF_HOME  # for backward compatibility. TODO: remove this in 1.0.0

把 default_home 修改为自己需要保存的文件路径即可。

我这里修改为:

python 复制代码
# default cache
default_home = os.path.join(os.path.expanduser("/root/autodl-tmp"), ".cache")
相关推荐
GPUStack17 天前
制作并量化GGUF模型上传到HuggingFace和ModelScope
大模型·huggingface·modelscope·genai·gguf
HyperAI超神经24 天前
对标Hugging Face?GitHub Models新增OpenAI o1/Llama 3.2等, 新功能支持模型并排比较
人工智能·机器学习·github·llama·huggingface
HackerTom1 个月前
safetensor存取pytorch模型参数、玄数据简例
pytorch·python·json·huggingface·safetensors
会飞的Anthony1 个月前
基于Python的自然语言处理系列(34):Huggingface 自定义分词器与数据集
人工智能·分词·huggingface
SiYuanFeng1 个月前
更改huggingface和pytorch的断点的默认下载位置
人工智能·pytorch·python·huggingface
Hiweir ·2 个月前
NLP任务之翻译
人工智能·python·深度学习·自然语言处理·huggingface·datasets
Hiweir ·2 个月前
NLP任务之文本分类(情感分析)
人工智能·自然语言处理·分类·huggingface
colorknight2 个月前
1.2.3 HuggingFists安装说明-MacOS安装
人工智能·低代码·macos·huggingface·数据科学·ai agent
Hiweir ·2 个月前
NLP任务之预测最后一个词
人工智能·gpt·自然语言处理·huggingface
colorknight2 个月前
1.2 HuggingFists安装说明-Linux安装
人工智能·低代码·llm·huggingface·数据科学·huggingfists·安装说明