krisGPT

/

student-tinybert-4layer-msra-ner_f1_76

like 0

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("token-classification", model="krisGPT/student-tinybert-4layer-msra-ner_f1_76")

# Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification

tokenizer = AutoTokenizer.from_pretrained("krisGPT/student-tinybert-4layer-msra-ner_f1_76")
model = AutoModelForTokenClassification.from_pretrained("krisGPT/student-tinybert-4layer-msra-ner_f1_76", device_map="auto")