Instructions to use deepset/roberta-base-squad2-distilled with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use deepset/roberta-base-squad2-distilled 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="deepset/roberta-base-squad2-distilled")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForQuestionAnswering tokenizer = AutoTokenizer.from_pretrained("deepset/roberta-base-squad2-distilled") model = AutoModelForQuestionAnswering.from_pretrained("deepset/roberta-base-squad2-distilled", device_map="auto") - Inference
- Notebooks
- Google Colab
- Kaggle
Example to train the model
#6
by shishirdwivedi2008 - opened
I am new to ML thing and want to use this image and train the model with custom pdf document so that model can answer question related to my custom pdf. Can somebody guide me how to train the model ?