gitextract_zinswnzk/ ├── README.md ├── all_prompt.md ├── configs/ │ ├── 2wikimultihopqa_llama3-8b-instruct.sh │ ├── 2wikimultihopqa_llama3.2-1b-instruct.sh │ ├── 2wikimultihopqa_qwen2.5-1.5b-instruct.sh │ ├── complexwebquestions_llama3-8b-instruct.sh │ ├── complexwebquestions_llama3.2-1b-instruct.sh │ ├── complexwebquestions_qwen2.5-1.5b-instruct.sh │ ├── hotpotqa_llama3-8b-instruct.sh │ ├── hotpotqa_llama3.2-1b-instruct.sh │ ├── hotpotqa_qwen2.5-1.5b-instruct.sh │ ├── popqa_llama3-8b-instruct.sh │ ├── popqa_llama3.2-1b-instruct.sh │ └── popqa_qwen2.5-1.5b-instruct.sh ├── prep_elastic.py ├── requirements.txt └── src/ ├── augment.py ├── encode.py ├── fewshot/ │ ├── 2wikimultihopqa.json │ └── hotpotqa.json ├── get_warmup_data.py ├── inference.py ├── prompt_template.py ├── retrieve/ │ ├── beir/ │ │ ├── .gitignore │ │ ├── .gitmodules │ │ ├── CONTRIBUTORS.txt │ │ ├── LICENSE │ │ ├── NOTICE.txt │ │ ├── README.md │ │ ├── beir/ │ │ │ ├── __init__.py │ │ │ ├── datasets/ │ │ │ │ ├── __init__.py │ │ │ │ ├── data_loader.py │ │ │ │ └── data_loader_hf.py │ │ │ ├── generation/ │ │ │ │ ├── __init__.py │ │ │ │ ├── generate.py │ │ │ │ └── models/ │ │ │ │ ├── __init__.py │ │ │ │ ├── auto_model.py │ │ │ │ └── tilde.py │ │ │ ├── logging.py │ │ │ ├── losses/ │ │ │ │ ├── __init__.py │ │ │ │ ├── bpr_loss.py │ │ │ │ └── margin_mse_loss.py │ │ │ ├── reranking/ │ │ │ │ ├── __init__.py │ │ │ │ ├── models/ │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── cross_encoder.py │ │ │ │ │ └── mono_t5.py │ │ │ │ └── rerank.py │ │ │ ├── retrieval/ │ │ │ │ ├── __init__.py │ │ │ │ ├── custom_metrics.py │ │ │ │ ├── evaluation.py │ │ │ │ ├── models/ │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── bpr.py │ │ │ │ │ ├── dpr.py │ │ │ │ │ ├── sentence_bert.py │ │ │ │ │ ├── sparta.py │ │ │ │ │ ├── splade.py │ │ │ │ │ ├── tldr.py │ │ │ │ │ ├── unicoil.py │ │ │ │ │ └── use_qa.py │ │ │ │ ├── search/ │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base.py │ │ │ │ │ ├── dense/ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── exact_search.py │ │ │ │ │ │ ├── exact_search_multi_gpu.py │ │ │ │ │ │ ├── faiss_index.py │ │ │ │ │ │ ├── faiss_search.py │ │ │ │ │ │ └── util.py │ │ │ │ │ ├── lexical/ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── bm25_search.py │ │ │ │ │ │ └── elastic_search.py │ │ │ │ │ └── sparse/ │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── sparse_search.py │ │ │ │ └── train.py │ │ │ └── util.py │ │ ├── examples/ │ │ │ ├── beir-pyserini/ │ │ │ │ ├── Dockerfile │ │ │ │ ├── config.py │ │ │ │ ├── dockerhub.sh │ │ │ │ └── main.py │ │ │ ├── benchmarking/ │ │ │ │ ├── benchmark_bm25.py │ │ │ │ ├── benchmark_bm25_ce_reranking.py │ │ │ │ └── benchmark_sbert.py │ │ │ ├── dataset/ │ │ │ │ ├── README.md │ │ │ │ ├── download_dataset.py │ │ │ │ ├── md5.csv │ │ │ │ └── scrape_tweets.py │ │ │ ├── generation/ │ │ │ │ ├── passage_expansion_tilde.py │ │ │ │ ├── query_gen.py │ │ │ │ ├── query_gen_and_train.py │ │ │ │ └── query_gen_multi_gpu.py │ │ │ └── retrieval/ │ │ │ ├── README.md │ │ │ ├── evaluation/ │ │ │ │ ├── README.md │ │ │ │ ├── custom/ │ │ │ │ │ ├── evaluate_custom_dataset.py │ │ │ │ │ ├── evaluate_custom_dataset_files.py │ │ │ │ │ ├── evaluate_custom_metrics.py │ │ │ │ │ └── evaluate_custom_model.py │ │ │ │ ├── dense/ │ │ │ │ │ ├── evaluate_ance.py │ │ │ │ │ ├── evaluate_bpr.py │ │ │ │ │ ├── evaluate_dim_reduction.py │ │ │ │ │ ├── evaluate_dpr.py │ │ │ │ │ ├── evaluate_faiss_dense.py │ │ │ │ │ ├── evaluate_sbert.py │ │ │ │ │ ├── evaluate_sbert_hf_loader.py │ │ │ │ │ ├── evaluate_sbert_multi_gpu.py │ │ │ │ │ ├── evaluate_tldr.py │ │ │ │ │ └── evaluate_useqa.py │ │ │ │ ├── late-interaction/ │ │ │ │ │ └── README.md │ │ │ │ ├── lexical/ │ │ │ │ │ ├── evaluate_anserini_bm25.py │ │ │ │ │ ├── evaluate_bm25.py │ │ │ │ │ └── evaluate_multilingual_bm25.py │ │ │ │ ├── reranking/ │ │ │ │ │ ├── README.md │ │ │ │ │ ├── evaluate_bm25_ce_reranking.py │ │ │ │ │ ├── evaluate_bm25_monot5_reranking.py │ │ │ │ │ └── evaluate_bm25_sbert_reranking.py │ │ │ │ └── sparse/ │ │ │ │ ├── evaluate_anserini_docT5query.py │ │ │ │ ├── evaluate_anserini_docT5query_parallel.py │ │ │ │ ├── evaluate_deepct.py │ │ │ │ ├── evaluate_sparta.py │ │ │ │ ├── evaluate_splade.py │ │ │ │ └── evaluate_unicoil.py │ │ │ └── training/ │ │ │ ├── train_msmarco_v2.py │ │ │ ├── train_msmarco_v3.py │ │ │ ├── train_msmarco_v3_bpr.py │ │ │ ├── train_msmarco_v3_margin_MSE.py │ │ │ ├── train_sbert.py │ │ │ └── train_sbert_BM25_hardnegs.py │ │ ├── setup.cfg │ │ └── setup.py │ ├── readme.md │ └── retriever.py ├── root_dir_path.py ├── utils.py └── warmup_lora.py