huggingface下载模型的最正确方法
zhangbo2008
编辑于 2023年12月28日 04:13

# transformsers 转onnx

import os

os.environ['HF_ENDPOINT&#​39;] = 'https://hf-mirror.com&#​39;

#加上这行之后又恢复以前的速度了!

from transformers import AutoModel, AutoConfig, AutoTokenizer

bert_model = AutoModel.from_pretrained('bert-base-uncased&#​39;)

config = AutoConfig.from_pretrained('bert-base-uncased&#​39;)

tokenizer = AutoTokenizer.from_pretrained('bert-base-uncased&#​39;)