Add MLM HEAD to pytorch_model.bin 08b6aa4
YoursSamson commited on
How to use OpenMOSS-Team/elasticbert-base with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="OpenMOSS-Team/elasticbert-base") # Load model directly
from transformers import AutoModelForMaskedLM
model = AutoModelForMaskedLM.from_pretrained("OpenMOSS-Team/elasticbert-base", device_map="auto")