protectai

/

bert-base-NER-onnx

like 4

Follow

Protect AI 99

tokenizer = AutoTokenizer.from_pretrained("protectai/bert-base-NER-onnx")
model = ORTModelForTokenClassification.from_pretrained("protectai/bert-base-NER-onnx")
ner = pipeline(
task="ner",
model=model,
tokenizer=tokenizer,
)

error:
AttributeError: 'ORTModelForTokenClassification' object has no attribute 'can_generate'