Mobilebert Squadv2

🚀 Mobile-Bert在Squad V2數據集上微調

本項目基於MobileBERT架構，適用於手持設備或資源有限的設備，可有效解決問答任務，為資源受限環境提供高效的問答解決方案。

🚀 快速開始

加載模型和分詞器

使用transformers庫，首先加載模型和分詞器。

from transformers import AutoModelForQuestionAnswering,  AutoTokenizer, pipeline

model_name = "aware-ai/mobilebert-squadv2"

model = AutoModelForQuestionAnswering.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

使用問答管道

qa_engine = pipeline('question-answering', model=model, tokenizer=tokenizer)
QA_input = {
    'question': 'your question?',
    'context': '. your context ................ '
}
res = qa_engine (QA_input)

📚 詳細文檔