huggingface-transformers - 如何更改拥抱脸转换器的默认缓存目录

Question

默认缓存目录是磁盘容量不足，我需要更改默认缓存目录的配置。

标签： huggingface-transformers

Answer 1

每次使用.from_pretrained加载模型时，您都可以通过设置参数来指定缓存目录cache_dir。您可以通过在每次使用之前（即在导入它之前！）库之前导出环境变量TRANSFORMERS_CACHE来定义默认位置。

蟒蛇的例子：

import os
os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'

bash 的示例：

export TRANSFORMERS_CACHE=/blabla/cache/