transformers/tests/models/gemma2
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 gemma 2 (#31659) 2024-06-27 17:36:19 +02:00
test_modeling_gemma2.py Iterative generation using Input embeds and past_key_values (#35890) 2025-02-06 11:06:05 +01:00