upload files 8e16e54
Dai Dai commited on
How to use xdai/mimic_longformer_base with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="xdai/mimic_longformer_base") # Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("xdai/mimic_longformer_base")
model = AutoModelForMaskedLM.from_pretrained("xdai/mimic_longformer_base", device_map="auto")