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
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