[generation] Less verbose warnings by default (#38179)

* tmp commit (imports broken)

* working version; update tests

* remove line break

* shorter msg

* dola checks need num_beams=1; other minor PR comments

* update early trainer failing on bad gen config

* make fixup

* test msg
This commit is contained in:
Joao Gante
2025-05-19 11:03:37 +01:00
committed by GitHub
parent 656e2eab3f
commit dbb9813dff
4 changed files with 251 additions and 247 deletions

View File

@@ -202,4 +202,4 @@ class Seq2seqTrainerTester(TestCasePlus):
data_collator=data_collator,
compute_metrics=lambda x: {"samples": x[0].shape[0]},
)
self.assertIn("The loaded generation config instance is invalid", str(exc.exception))
self.assertIn("Fix these issues to train your model", str(exc.exception))