Yih-Dar ab98f0b0a1 avoid calling gc.collect and cuda.empty_cache (#34514) ...

* update

* update

* update

* update

* update

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

2024-10-31 16:36:13 +01:00

..

__init__.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00

test_modeling_rag.py

avoid calling gc.collect and cuda.empty_cache (#34514)

2024-10-31 16:36:13 +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