transformers/tests/models/rag
Yih-Dar 2cc3cc835f
Add `dataset_revision` argument to `RagConfig` (#29610)
* add arg

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2024-03-14 16:48:11 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_rag.py Add `dataset_revision` argument to `RagConfig` (#29610) 2024-03-14 16:48:11 +01:00
test_modeling_tf_rag.py Add `dataset_revision` argument to `RagConfig` (#29610) 2024-03-14 16:48:11 +01:00
test_retrieval_rag.py Disallow `pickle.load` unless `TRUST_REMOTE_CODE=True` (#27776) 2023-12-04 16:48:37 +01:00
test_tokenization_rag.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00