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

GenerationConfig validate both constraints and force_words_ids (#29163)

Browse Source 
GenerationConfig validate both options for constrained decoding: constraints and force_words_ids
This commit is contained in:
FredericOdermatt
2024-02-27 09:43:52 +09:00
committed by GitHub
parent 3fcfbe7549
commit 871ba71dfa
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/generation/test_configuration_utils.py
View File

@@ -156,6 +156,11 @@ class GenerationConfigTest(unittest.TestCase):
# Impossible sets of contraints/parameters will raise an exception
with self.assertRaises(ValueError):
GenerationConfig(do_sample=False, num_beams=1, num_return_sequences=2)
with self.assertRaises(ValueError):
# dummy constraint
GenerationConfig(do_sample=True, num_beams=2, constraints=["dummy"])
with self.assertRaises(ValueError):
GenerationConfig(do_sample=True, num_beams=2, force_words_ids=[[[1, 2, 3]]])
# Passing `generate()`-only flags to `validate` will raise an exception
with self.assertRaises(ValueError):