transformers/tests/models/gpt_bigcode
Yaswanth Gali 7aee036e54
Iterative generation using Input embeds and past_key_values (#35890)
* Iterative generation using input embeds

* ruff fix

* Added Testcase

* Updated comment

* ♻️ Refactored testcase

* Skip test for these models

* Continue generation using input embeds and cache

* Skip generate_continue_from_embeds test

* Refactor `prepare_input_for_generation` func

* Continue generation using input embeds and cache

* Modular changes fix

* Overwrite 'prepare_inputs_for_generation' function
2025-02-06 11:06:05 +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 Iterative generation using Input embeds and past_key_values (#35890) 2025-02-06 11:06:05 +01:00