Instructions to use castorini/tct_colbert-v2-msmarco with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use castorini/tct_colbert-v2-msmarco with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="castorini/tct_colbert-v2-msmarco")# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("castorini/tct_colbert-v2-msmarco") model = AutoModel.from_pretrained("castorini/tct_colbert-v2-msmarco") - Notebooks
- Google Colab
- Kaggle
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
This model is to reproduce a variant of TCT-ColBERT-V2 dense retrieval models described in the following paper:
Sheng-Chieh Lin, Jheng-Hong Yang, and Jimmy Lin. In-Batch Negatives for Knowledge Distillation with Tightly-CoupledTeachers for Dense Retrieval. RepL4NLP 2021.
You can find our reproduction report in Pyserini here.
- Downloads last month
- 20