我的博客
欢迎来到我的博客
bunny.icu

解决Huggingface被墙下载模型的问题

解决Huggingface被墙下载模型的问题

解决Huggingface被墙下载模型的问题

pip install -U huggingface_hub
# 只在当前终端生效,如果想长期生效改source文件
export HF_ENDPOINT=https://hf-mirror.com
huggingface-cli download --resume-download utrobinmv/t5_translate_en_ru_zh_large_1024

这样就会下载模型到本地,然后可以从本地加载

from transformers import T5ForConditionalGeneration, T5Tokenizer

print("t5_translate_en_ru_zh_large_1024 模型加载中...")
device = 'cuda'

model_name = 'utrobinmv/t5_translate_en_ru_zh_large_1024'
model = T5ForConditionalGeneration.from_pretrained(model_name)
model.to(device)
tokenizer = T5Tokenizer.from_pretrained(model_name)
# ...

Reference

https://blog.csdn.net/weixin_43303286/article/details/134342476

版权声明


本作品系原创, 转载须遵循 CC BY-NC-ND 4.0 许可协议
本文标题:解决Huggingface被墙下载模型的问题
本文链接:https://www.bunny.icu/archives/1785

推荐文章

发表评论

textsms
account_circle
email

bunny.icu

解决Huggingface被墙下载模型的问题
参考 https://blog.csdn.net/weixin_43303286/article/details/134342476
扫描二维码继续阅读
2024-04-25