AutoRefine / download_model.py
yrshi's picture
update some files
0fe2213
raw
history blame
185 Bytes
from transformers import AutoTokenizer, AutoModel
model_name = 'intfloat/e5-base-v2'
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModel.from_pretrained(model_name)