transformers/tests/models/ctrl
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_ctrl.py avoid calling gc.collect and cuda.empty_cache (#34514) 2024-10-31 16:36:13 +01:00
test_modeling_tf_ctrl.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_tokenization_ctrl.py Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00