SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Generate: legacy mode is only triggered when generation_config is untouched (#25962)

Browse Source
This commit is contained in:
Joao Gante
2023-09-12 12:08:17 +01:00
committed by GitHub
parent 18abc756c5
commit 3319eb5490
6 changed files with 53 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/generation/test_utils.py
View File

@@ -2880,7 +2880,7 @@ class GenerationIntegrationTests(unittest.TestCase, GenerationIntegrationTestsMi
# Generation config max_length != 20 -> no warning
with warnings.catch_warnings(record=True) as warning_list:
# generation_config is modified -> legacy mode is disabled = generation_config takes precedence
model.generation_config.max_length = 10
model.generation_config._from_model_config = False # otherwise model.config.max_length=20 takes precedence
model.generate(input_ids)
self.assertEqual(len(warning_list), 0)