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
test_modeling_rag.py
test_modeling_tf_rag.py
test_retrieval_rag.py
test_tokenization_rag.py