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

帅老头123

已于 2024-11-27 16:58:51 修改

阅读量230

点赞数 2

CC 4.0 BY-SA版权

文章标签：深度学习 pytorch 人工智能

于 2024-11-27 16:57:51 首次发布

本文链接：https://blog.youkuaiyun.com/zhangning/article/details/144089297

在别在模块加载之前设置，或者在环境变量里设置


import os
os.environ['XDG_CACHE_HOME'] = "D:/_torch-cache" # os.getcwd()
os.environ['HF_HOME'] = "D:/_torch-cache/huggingface"
os.environ['HF_HUB_CACHE'] = "D:/_torch-cache/huggingface/hub"
os.environ['TORCH_HOME'] = "D:/_torch-cache/torch" # os.getcwd()
os.environ['TRANSFORMERS_CACHE'] = "D:/_torch-cache/huggingface/hub"

from transformers import (pipeline)
import torchvision.models as models
。。。

查了很多文档说的很不准确，跟踪了一下代码才找全