tests/pipelines/test_pipelines_text_generation.py · 636b03244cb3c5bac6d12a5a968d5024e0fde7c3 · chenpangpang / transformers

️ 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