Using
Hi thanks for your work, is there an example of using and downloading this model for text encoding?
I try to use it with
from transformers import AutoTokenizer, AutoModelForPreTraining
tokenizer = AutoTokenizer.from_pretrained("Vuy/rubert-tiny2-onnx")
model = AutoModelForPreTraining.from_pretrained("Vuy/rubert-tiny2-onnx")
But have error: Vuy/rubert-tiny2-onnx does not appear to have a file named pytorch_model.bin, tf_model.h5, model.ckpt or flax_model.msgpack.
пожалуйста, отзовись
Привет. Эта модель для пакета optimum.
Я использую ее так:
from optimum.onnxruntime import ORTModelForFeatureExtraction
from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = ORTModelForFeatureExtraction.from_pretrained(model_name)