transformers/tests/models/rag
Yih-Dar 1d63b0ec36
Disallow pickle.load unless TRUST_REMOTE_CODE=True (#27776)
* fix

* fix

* Use TRUST_REMOTE_CODE

* fix doc

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2023-12-04 16:48:37 +01:00
..
__init__.py
test_modeling_rag.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_modeling_tf_rag.py Big TF test cleanup (#24282) 2023-06-16 15:40:49 +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