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

fix typo in test xlm tf

Browse Source
This commit is contained in:
Patrick von Platen
2020-03-09 11:34:31 +01:00
parent 4620caa864
commit b73dd1a0e4
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_modeling_tf_xlm.py
View File

@@ -341,5 +341,5 @@ class TFXLMModelLanguageGenerationTest(unittest.TestCase):
447, 447,
] # the president the president the president the president the president the president the president the president the president the president ] # the president the president the president the president the president the president the president the president the president the president
# TODO(PVP): this and other input_ids I tried for generation give pretty bad results. Not sure why. Model might just not be made for auto-regressive inference # TODO(PVP): this and other input_ids I tried for generation give pretty bad results. Not sure why. Model might just not be made for auto-regressive inference
output_ids = model.generate(input_ids) output_ids = model.generate(input_ids, do_sample=False)
self.assertListEqual(output_ids[0].numpy().tolist(), expected_output_ids, do_sample=False) self.assertListEqual(output_ids[0].numpy().tolist(), expected_output_ids)