默认缓存目录磁盘容量不足,我需要更改默认缓存目录的配置。
您可以在每次加载模型时指定缓存目录.from_pretrained https://huggingface.co/transformers/main_classes/model.html#transformers.PreTrainedModel.from_pretrained通过设置参数cache_dir
。您可以通过导出环境变量来定义默认位置TRANSFORMERS_CACHE https://huggingface.co/transformers/installation.html?highlight=transformers_cache#caching-models每次使用之前(即before导入它!)库)。
蟒蛇的例子:
import os
os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
bash 的示例:
export TRANSFORMERS_CACHE=/blabla/cache/
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)