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)

📚 ドキュメント