First model version b247ff3
Zhou Yucheng commited on
How to use 9pinus/macbert-base-chinese-medical-collation with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="9pinus/macbert-base-chinese-medical-collation") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("9pinus/macbert-base-chinese-medical-collation")
model = AutoModelForTokenClassification.from_pretrained("9pinus/macbert-base-chinese-medical-collation", device_map="auto")