rajpurkar/squad
Viewer • Updated • 98.2k • 214k • 369
How to use Dylan1999/bert-squad-mrc with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("question-answering", model="Dylan1999/bert-squad-mrc") # Load model directly
from transformers import AutoTokenizer, AutoModelForQuestionAnswering
tokenizer = AutoTokenizer.from_pretrained("Dylan1999/bert-squad-mrc")
model = AutoModelForQuestionAnswering.from_pretrained("Dylan1999/bert-squad-mrc", device_map="auto")