tests/pipelines/test_pipelines_text_generation.py · abf8f54a019ce14b5eaffa68c6dd883be13fe66e · chenpangpang / transformers

"tests/models/distilbert/test_modeling_distilbert.py" did not exist on "783a61699962f4b058688db21d417e1932423417"

️ Raise `Exception` when trying to generate 0 tokens

Daniel Korat authored Feb 07, 2024



* change warning to exception

* Update src/transformers/generation/utils.py
Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* validate `max_new_tokens` > 0 in `GenerationConfig`

* fix truncation test parameterization in `TextGenerationPipelineTests`

---------
Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

abf8f54a

test_pipelines_text_generation.py 14.7 KB

Replace test_pipelines_text_generation.py