"docs/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "97f9d8a992749584455fa76822956190a75c64f3"
Commit b73dd1a0 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

fix typo in test xlm tf

parent 4620caa8
......@@ -341,5 +341,5 @@ class TFXLMModelLanguageGenerationTest(unittest.TestCase):
447,
] # 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
output_ids = model.generate(input_ids)
self.assertListEqual(output_ids[0].numpy().tolist(), expected_output_ids, do_sample=False)
output_ids = model.generate(input_ids, do_sample=False)
self.assertListEqual(output_ids[0].numpy().tolist(), expected_output_ids)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment