🚨🚨🚨 [pipelines] update defaults in pipelines that can generate (#38129)

* pipeline generation defaults

* add max_new_tokens=20 in test pipelines

* pop all kwargs that are used to parameterize generation config

* add class attr that tell us whether a pipeline calls generate

* tmp commit

* pt text gen pipeline tests passing

* remove failing tf tests

* fix text gen pipeline mixin test corner case

* update text_to_audio pipeline tests

* trigger tests

* a few more tests

* skips

* some more audio tests

* not slow

* broken

* lower severity of generation mode errors

* fix all asr pipeline tests

* nit

* skip

* image to text pipeline tests

* text2test pipeline

* last pipelines

* fix flaky

* PR comments

* handle generate attrs more carefully in models that cant generate

* same as above
This commit is contained in:
Joao Gante
2025-05-19 18:02:06 +01:00
committed by GitHub
parent 6f9da7649f
commit 9c500015c5
23 changed files with 361 additions and 438 deletions

View File

@@ -86,6 +86,7 @@ class DocumentQuestionAnsweringPipelineTests(unittest.TestCase):
image_processor=image_processor,
processor=processor,
torch_dtype=torch_dtype,
max_new_tokens=20,
)
image = INVOICE_URL