mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-23 22:38:58 +06:00
9ec0f01b6c
* initial * code quality test * code quality * added test functions in test_modeling_rag.py and test_retrieval_rag.py to test end2end retreiver * minor change in test_modeling_rag * fixed tests * Update examples/research_projects/rag-end2end-retriever/README.md typo corrected as suggested by lhoestq Co-authored-by: Quentin Lhoest <42851186+lhoestq@users.noreply.github.com> * Update examples/research_projects/rag-end2end-retriever/finetune_rag.py type change suggested by lhoestq Co-authored-by: Quentin Lhoest <42851186+lhoestq@users.noreply.github.com> * Update src/transformers/models/rag/retrieval_rag.py Adding this change as mentioned by lhoestq. Co-authored-by: Quentin Lhoest <42851186+lhoestq@users.noreply.github.com> * completed the minor changes suggested by the reviewers Co-authored-by: Quentin Lhoest <42851186+lhoestq@users.noreply.github.com>
8 lines
135 B
Plaintext
8 lines
135 B
Plaintext
faiss-cpu >= 1.7.0
|
|
datasets >= 1.6.2
|
|
psutil >= 5.7.0
|
|
torch >= 1.4.0
|
|
pytorch-lightning == 1.3.1
|
|
nvidia-ml-py3 == 7.352.0
|
|
ray >= 1.3.0
|