使用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")