Instructions to use nvidia/dragon-multiturn-query-encoder with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use nvidia/dragon-multiturn-query-encoder with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="nvidia/dragon-multiturn-query-encoder")# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("nvidia/dragon-multiturn-query-encoder") model = AutoModel.from_pretrained("nvidia/dragon-multiturn-query-encoder") - Inference
- Notebooks
- Google Colab
- Kaggle
Commands for running evaluation
python evaluate.py --eval-dataset doc2dial
python evaluate.py --eval-dataset quac
python evaluate.py --eval-dataset qrecc
For the evaluations of topiocqa and inscit, the Wikipedia corpora need to be downloaded from their original repositories.
- topiocqa: https://github.com/McGill-NLP/topiocqa
- inscit: https://github.com/ellenmellon/INSCIT