【极简】huggingface如何用离线方式2025.1.22

Yonggie

于 2025-01-22 10:24:24 发布

阅读量427

点赞数 4

文章标签：深度学习人工智能 pytorch python 算法

本文链接：https://blog.csdn.net/Yonggie/article/details/145298285

版权

根据huggingface document：

要么启动命令行上加上标志，要么python加载时候带上local_files_only=True。

命令行：

HF_DATASETS_OFFLINE=1 HF_HUB_OFFLINE=1 \
python examples/pytorch/translation/run_translation.py --model_name_or_path google-t5/t5-small --dataset_name wmt16 --dataset_config ro-en ...

python

from transformers import T5Model

model = T5Model.from_pretrained("./path/to/local/directory", local_files_only=True)

模型在国内可以通过huggingface mirror下载。https://hf-mirror.com/。