Instructions to use aware-ai/mobilebert-squadv2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use aware-ai/mobilebert-squadv2 with Transformers:
# Use a pipeline as a high-level helper # Warning: Pipeline type "question-answering" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # pip install "transformers<5.0.0" from transformers import pipeline pipe = pipeline("question-answering", model="aware-ai/mobilebert-squadv2")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForQuestionAnswering tokenizer = AutoTokenizer.from_pretrained("aware-ai/mobilebert-squadv2") model = AutoModelForQuestionAnswering.from_pretrained("aware-ai/mobilebert-squadv2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
-
Create README.md
# Mobile-Bert fine-tuned on Squad V2 dataset This is based on mobile bert architecture suitable for handy devices or device with low resources. ## usage using transformers library first load model and Tokenizer ``` from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline model_name = "aware-ai/mobilebert-squadv2" model = AutoModelForQuestionAnswering.from_pretrained(model_name) tokenizer = AutoTokenizer.from_pretrained(model_name) ``` use question answering pipeline ``` qa_engine = pipeline('question-answering', model=model, tokenizer=tokenizer) QA_input = { 'question': 'your question?', 'context': '. your context ................ ' } res = qa_engine (QA_input) ```78873df -
Adding `safetensors` variant of this model (#1)
- Adding `safetensors` variant of this model (f04e8002cfd0c37f71ef3a8e048fb3072b3d758a) Co-authored-by: Safetensors convertbot <SFconvertbot@users.noreply.huggingface.co>4bf184a
-
Update saved_model.tar.gz
5df6060
-
Update training_args.bin
2293e81 -
Update vocab.txt
02a15e8 -
Update nbest_predictions_test.json
004ddea -
Update null_odds_test.json
ecaaf0a -
Update config.json
64f07a5 -
Update special_tokens_map.json
874aa64 -
Update tokenizer_config.json
d34f671 -
Update predictions_test.json
7f0d418 -
Update pytorch_model.bin
2a99db3 -
Update model_args.json
ff543a1 -
initial commit
efd79c8