transformers/tests/models/gpt_bigcode
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 Add GPTBigCode model (Optimized GPT2 with MQA from Santacoder & BigCode) (#22575) 2023-04-10 10:57:21 +02:00
test_modeling_gpt_bigcode.py avoid calling gc.collect and cuda.empty_cache (#34514) 2024-10-31 16:36:13 +01:00