huggingface离线模式及默认保存路径

T5Tokenizer.from_pretrained()函数会在线下载huggingface上的模型,并自动保存在C盘的.cache路径下:

C:\Users\xxxxx\.cache\huggingface\hub\models--google--t5-v1_1-base
# 这里xxxxx代表自己电脑用户名

huggingface离线下载

huggingface离线模式及默认保存路径_第1张图片
'google/t5-v1_1-base'为例,几个json文件全要下载,如果是tensorflow下载tf_model.h5,如果是pytorch下载pytorch_model.bin,然后把这些文件放在上面的路径下面。再在代码中修改路径

# DEFAULT_T5_NAME = 'google/t5-v1_1-base'      # 原代码
DEFAULT_T5_NAME = r'C:\Users\xxxxx\.cache\huggingface\hub\models--google--t5-v1_1-base'   # 修改为自己的路径

你可能感兴趣的:(深度学习)