transformers icon indicating copy to clipboard operation
transformers copied to clipboard

Add support for XTR

Open mjeensung opened this issue 7 months ago • 4 comments

Model description

XTR (ConteXtualized Token Retriever) is a multi-vector retrieval model that improves efficiency by focusing on retrieving and ranking the most important document tokens. Details are described at https://arxiv.org/abs/2304.01982.

Open source status

  • [X] The model implementation is available
  • [X] The model weights are available

Provide useful links for the implementation

Implementation for PyTorch: https://github.com/mjeensung/xtr-pytorch Weight: https://huggingface.co/google/xtr-base-en Author: @jhyuklee

mjeensung avatar Jul 09 '24 22:07 mjeensung