transformers/tests/models/rag
Yih-Dar 224bde91ca
Avoid GPU OOM for a TF Rag test (#17638)
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-06-10 18:50:29 +02:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_rag.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_tf_rag.py Avoid GPU OOM for a TF Rag test (#17638) 2022-06-10 18:50:29 +02:00
test_retrieval_rag.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_tokenization_rag.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00