transformers/tests/models/mistral
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 [Mistral] Mistral-7B-v0.1 support (#26447) 2023-09-27 18:30:46 +02:00
test_modeling_flax_mistral.py Flax mistral (#26943) 2024-01-31 14:19:02 +01:00
test_modeling_mistral.py avoid calling gc.collect and cuda.empty_cache (#34514) 2024-10-31 16:36:13 +01:00
test_modeling_tf_mistral.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00